Cleanup code (using flake8).