update generate.py