Edit the model classes