Add Black for code formatting

[gaurav]

It would be great to add a code formatter like Black to this repo so that our code here is consistenly formatted and therefore hopefully easier to read.