Added probability_new