Fix all remaining flake8 errors, module wide