modify tree so that training is standalone
modify tree so that training is standalone