Add generator.py