Main updates to model selection script