Should update this docstring too