Fixed flake8 errors