Attempted to get probabilities right