Fixed bugs in the probability module and its viz.