Fix: flake8 tests
Merge pull request #25 from william-p/flake8

Fix: flake8 tests