add generator.py