Update template_gen.py