Changed how the model is updated