Created prediction entity class. need mappings