Pep8

Mostly the code was fine, few missed things:

- 2 white lines between imports and class def
- 2 white lines in methods not in class
- 2 spaces before inline comments