Tree UI improvements.