Add functions to return predicts