flake8 cleanup.