Updated Generator.py

* Added a function to the file, commented out, after taking advice from
George's friend
* Fixed indentation issues a-fudging-gain