trying to fix flake8 errors