please update the docstring, too