flake8 warnings fixed