Refactor
Need to start the tree generation