Remove Probability class