predict_prob now scales probabilities properly.