flake8 corrections