update to model class