Generator.py works as expected now