Add the model class