Tidy with flake8