Improve the tree style