Fixing flake8