adding our own base model