Changed Node to TreeNode