Add tree type annotation