Add comprehensive documentation including docstrings

[Neihouse]

This task involves adding comprehensive documentation and docstrings to the codebase to make it easier for other developers to understand and contribute.