make tree branch aware
type
Merge pull request #2 from earlearlearl/master

Make tree aware of current branch