JZ: Add docstrings everywhere

Except for files that will be factored out into a separate repo