Added remove_node function