Add base model class