Fixed flake8 issues'