Add docstrings for all functions