modify Predict func