Added category probabilities tests