add probability computation