Removed flake8 warnings.