Update new add_node() and remove_node()