Fixing couple of PEP8 errors.
Merge pull request #5 from koddsson/fix-pep8

Fixing couple of PEP8 errors.