Add the rest of the probability example