Rename segment to predict_next