Add TrainedModel class; implement predict() for it