add model selection stuff