Added Node class to the tree walker