[MOD] Fixing flake8 errors