Added probability computation