docstring improved