Add example of `return_model` to the docstrings.