Finally adding train model.