Add docstrings to all functions