text: Title: Ability to add custom imports
  Body: **Is your feature request related to a problem? Please describe.**
  Can't customize my template to add new imports
  
  **Describe the solution you'd like**
  New argument `additional_imports` in `generate` function which adds additional imports to final rendered template
  
  **Describe alternatives you've considered**
  -
  
  **Additional context**
  -