Changed return type docstring