Add support for booster.model() to get model only