added generator.py to repo