Minor refactor to be more PEP8
Merge pull request #2 from caretdashcaret/pep8

Minor refactor to be more PEP8