Added testing for tree