DOC: Update docstrings