Address review comment: Better docstring.