To solve/ Generator.py