Correct information gain call (I think) - repairing code after Anu and Surhud's alternative information suggestion
Tidying up