Refactor tree branching code
Refactor tree branching code

bhavanaananda/DataStage,dataflow/DataStage,dataflow/DataStage,bhavanaananda/DataStage,bhavanaananda/DataStage,bhavanaananda/DataStage,bhavanaananda/DataStage,dataflow/DataStage