add Prediction function