flake8 fixes
Merge pull request #41 from nir0s/flake8-fixes

flake8 fixes