Removed flake8 violations