Needed to return additional information about the predictions.