Fixing flake8 errors