Update the Model class