fixing flake8 problems