Shall we add it? Or just change the docstring?