Fixes #7 pep8 compliance