Updates to Prediction classes