Update Node and Tree class