Updated to pass flake8 code checks.