Limited Tree Functionality

[sushantnair]

As can be seen in lines 135-144 (as of 27th March 2024), the code can do only two levels of the tree. If you can find a way to make the tree go deeper and deeper until an appropriate stopping condition is arrived at, please contribute!
Of course the sample Dataset I've provided needed only two levels of tree and so looks neat, but I want the code to be a generalized code that can run on any dataset..