Update the docs to encourage returning the context in the model hook