Add a base model for mocking