cleaned the repo using flake8

'
@tessie Oops, I didn't catch this until I tried to run it. We do need this import, so please add it back in. I will wait for that fox before merging.