PEP8 cleanup
Fixes #28