added new train model