Started coding the information gain function