Description
Improve the documentation and understanding of the codebase by adding docstrings to all functions. We should pick a specific docstring style, like Google Style Guide and stick to it. This will provide valuable information for developers working with the code.