Add function docstrings