probability update

Probability now works.