Fixing decision tree buiding