Fix flake8 warnings
Merge pull request #3 from chromano/master

Fix flake8 warnings