Ran script through flake8, fixed issues
Ran script through flake8, fixed issues