Use black formatting

[anna-hope]

As black is becoming the de-facto standard formatter for Python code, I think we should use it in this repository. The formatting is wonky in multiple places, and using black would make the code more readable.