Fix up flake8 errors