Adding the Node class also