Fixed AI4_1 Probability bug
Was not correctly updating probabilities occasionally. Fixed now