Fixed up some PEP8 issues.

Fixed only minor PEP8 style errors. Still have more to fix like lines
that runs over 80 characters.
Cleaned up code further.

Cleaned up code further and converted prints and inputs to Python3.
Merge pull request #5 from JoelMon/pep8

Pep8