Added return info to the docstring