improve docstring