Fixed some flake8 issues