PEP8 cleanup
Merge pull request #13 from dionyziz/cleanup

PEP8 cleanup