Fixed flake8 warnings.