fix flake8 warnings