Get flake8 to pass