Add documentation

[ScottWales]

Since there's only a few functions, probably no need to use readthedocs or anything

• [ ] Convert the function comments to docstrings
• [ ] Describe the intended workflow in the readme file

[aidanheerdegen]

• [ ] Add scientific reference for method to README