Modified the way max info gain is computer