Update model classes