Clean up for flake8