Using flake8 to tidy up a file