clean tree struct code
Merge pull request #3 from xiangli-cmu/master

clean up tree.go