Add display tree functionality

not finished