Update genrerator.py

cleaned up main a lil bit