Add successProbability function