update gen.py