Fixing the Node class