Fix flake8 warnings

This helps ensuring the best possible start
Merge pull request #3 from OdyX/fix-flake8-warnings

Fix flake8 warnings