Continued with working on the tree implementation and doing the maximum information gain.
Also wrote tests to check the information gain works properly.