Merge pull request #20 from shamenchens/TreeData-FixTreeTotal

Fix tree total size
refactor tree total size