PEP 8 compliance
Merge pull request #4 from yoloseem/master

PEP 8 compliance