added train model dir