Fixed flake8 issues