Updated to pick best model