Argument 'prob' -> 'probs'