add model selection