added train model