Add remove function for rb_tree