cleaned up the flake8 stuff