tests passing
fix flake8