updated model selection section