Generator.py changed