fixing flake8 tests
Merge pull request #9 from vmahuli/master

fixing flake8 tests