fixed issues in probability.doctest to make it runnable