Added probability tests