Added more tests for the base model.