Improve docstring

Thanks, @cyli!