Made the tree generator take into account only one leaf