predict_probability changed return types... very tricky