fix pep8 styling
Merge pull request #2 from jivid/master

Fix PEP8 styling