add GetTreeCount func