Improve docstring

[ci skip]