fixed query, got it wrong
added information gain, still need to compute it