pep8 cleanup in main.py

main.py is now pep8 friendly!