Add gen.py