Modified the parent model class