made some changes to make only 2d trees for now