Update probability calculations