fixed up flake8 fails