still struggling with pruning tree