Improve tree node