Use sum() to calculate probabilities instead of len()