Minor fixes to decision tree learning