Modified the model class