flake8 cleanup