Update model class