Modified the node class