Improve docstring