Refactor init_tree_grow() to record the tree size
Add ability to parse input set nodes