cleaned up flake8 warnings