Updated gen.py