updated gen.py