added train model process