Updates to decision trees notebook