Implemented Prediction entity and repository class