Update docstring accordingly