Minor update
Add method to print tree