Adding BaseModel class