MPI Tree now works with the pruned trees