Add model definition for Prediction