added code for model selection on repo