Flake8 complicance

These changes have not been tested, it might have broken something.
Merge pull request #2 from torbjorntorbjorn/flake8

Flake8 complicance