Update generate.py