Fixed bugs in ProbModel and InfoGain