flake8 fixes