Minor changes to visit Tree