Updates

* Fixed indentation errors in generator.py
* Added comments to main.py