Fixed some flake8 errors