Improve: Add docstrings to functions