Added display tree functionality