Added the return model to the puts.