Started on decision tree prediction stuff