whitespace and pep8 compliance :heart:

I herd u liek pep8 so I put some style in your code.