Clean up code styling after running flake8