Added in some probability stuff (finally)