Add remove_node function.