adding train model py files