docstring improvement