Add remove_node method