Update Prediction.h