[DOC] format/add docstrings