I think that you want to add a comment here on the purpose of 'return_models'.