style fixes
Merge pull request #8 from ralsina/flake8

style fixes