Pep8 stuff

Nothing significant, just made it look prettier