Create generator.py