Fix pep8 errors
Merge pull request #5 from jpvanhal/linting

Fix pep8 errors