Fix pep8 warnings
Merge pull request #2 from abelloso/issue/1

Fix pep8 warnings