Fix flake8 errors
Merge pull request #7 from guewen/master-pep8

Fix flake8 errors