Made code PEP8 compliant
Merge pull request #3 from stormbreakerbg/master

PEP8 compliant tests