Improve formatting of tree output