Updated Generator.py