improving the probabilities section