Added code for baby-trees post-processing