Add test for trees