Review: improve code quality

[cthoyt]

The code in this repository doesn't appear to have consistent styling. This can be easily addressed with black, which reformats code in a deterministic way. It's not perfect in how it looks, but at least it makes it more standard for others to read.

Further, you should use tools like flake8 and pylint to find other common mistakes/pitfalls in the way the code is written. Keep in mind that your code is effectively the methods section of your manuscript, and that it should be meant to be read by all users.