Use Trait or extends to the BaseModel