Update generator.py

to check