Made TreeNode in section 22 match code