Some bugs in information gain fixed