Improve tree type