Add drawTree util