Fixed a couple of bugs with the decision tree