Is your feature request related to a problem? Please describe.
Some of the public facing docstrings aren't very comprehensive and could probably do with a bit of an upgrade
Describe the solution you'd like
Improve the documentation so all public facing features have appropriate docstrings, return types, type hints, etc.