Added probability calculations