Added ability to select model in model output