Modified set_main_model to only accept one model instead of an array of models