changes to the model classes