and one more time, I agree :-) Just add a few comments here and there in a similar way as you have done in `prune_tree_machine`.