Update Gen.py