updated the model classes