Fixed flake8 linting