Pep8

Cleaned up the code some and converted print and inputs to Python3.