Tidy with flake8
Merge pull request #2 from funkybob/master

Tidy with flake8