predict_probability changed return types... why are there so many ostriches? why?