Added generator.py file