Update gen.py