adjust the API for model selection