@@ -23,5 +23,6 @@
 
 - (instancetype) initWithValue:(id)value;
 - (void) addChild:(TreeNode *)newChild;
+- (void) removeTreeNode:(TreeNode *)treeNode;
 - (NSArray<TreeNode *> *) visibleNodes;
 @end
@@ -44,6 +44,28 @@ - (void) addChild:(TreeNode *)newChild {
     [_children addObject:newChild];
 }
 
+- (void) removeTreeNode:(TreeNode *)treeNode {
+    [self _internalRemove:treeNode];
+}
+
+- (BOOL) _internalRemove:(TreeNode *)node {
+    BOOL result = NO;
+    for (TreeNode *child in _children) {
+        if (node == child) {
+            [_children removeObject:node];
+            result = YES;
+            break;
+        }
+        if (child.isFolder) {
+            result = [child _internalRemove:node];
+            if (result) {
+                break;
+            }
+        }
+    }
+    return result;
+}
+
 - (NSUInteger) levelDepth {
     NSUInteger cnt = 0;
     if (_parent != nil) {
 
@@ -17,6 +17,7 @@
 - (NSInteger) numberOfRowsInTreeView:(UITreeView *)treeView;
 - (TreeNode *) treeView:(UITreeView *)treeView treeNodeForRow:(NSInteger)row;
 - (NSInteger) treeView:(UITreeView *)treeView rowForTreeNode:(TreeNode *)treeNode;
+- (void) treeView:(UITreeView *)treeView removeTreeNode:(TreeNode *)treeNode;
 
 //@optional
 - (void) treeView:(UITreeView *)treeView didSelectForTreeNode:(TreeNode *)treeNode;
 
@@ -86,9 +86,17 @@ - (BOOL) tableView:(UITableView *)tableView canEditRowAtIndexPath:(NSIndexPath *
 }
 
 - (void) tableView:(UITableView *)tableView commitEditingStyle:(UITableViewCellEditingStyle)editingStyle forRowAtIndexPath:(NSIndexPath *)indexPath {
+    TreeNode *treeNode = [self treeNodeForIndexPath:indexPath];
     if (editingStyle == UITableViewCellEditingStyleDelete) {
-        // Delete the row from the data source
-        [tableView deleteRowsAtIndexPaths:@[indexPath] withRowAnimation:UITableViewRowAnimationFade];
+        if ([_treeViewDelegate respondsToSelector:@selector(treeView:removeTreeNode:)]) {
+            [_treeViewDelegate treeView:self removeTreeNode:treeNode];
+        }
+        if (treeNode.isFolder && treeNode.expanded && treeNode.hasChildren) {
+            [self reloadData];
+            [self resetSelection];
+        } else {
+            [tableView deleteRowsAtIndexPaths:@[indexPath] withRowAnimation:UITableViewRowAnimationFade];
+        }
     } else if (editingStyle == UITableViewCellEditingStyleInsert) {
         // Create a new instance of the appropriate class, insert it into the array, and add a new row to the table view
     }
 
@@ -31,6 +31,10 @@ - (NSInteger) treeView:(UITreeView *)treeView rowForTreeNode:(TreeNode *)treeNod
     return [[_rootTreeNode visibleNodes] indexOfObject:treeNode];
 }
 
+- (void) treeView:(UITreeView *)treeView removeTreeNode:(TreeNode *)treeNode {
+    [_rootTreeNode removeTreeNode:treeNode];
+}
+
 - (void) treeView:(UITreeView *)treeView didSelectForTreeNode:(TreeNode *)treeNode {
     NSLog(@"Node %@ selected", treeNode.title);
 }
Original file line number	Diff line number	Diff line change
`@@ -31,6 +31,10 @@ - (NSInteger) treeView:(UITreeView )treeView rowForTreeNode:(TreeNode )treeNod`
`31`	`31`	`return [[_rootTreeNode visibleNodes] indexOfObject:treeNode];`
`32`	`32`	`}`
`33`	`33`
	`34`	`+- (void) treeView:(UITreeView )treeView removeTreeNode:(TreeNode )treeNode {`
	`35`	`+ [_rootTreeNode removeTreeNode:treeNode];`
	`36`	`+}`
	`37`	`+`
`34`	`38`	`- (void) treeView:(UITreeView )treeView didSelectForTreeNode:(TreeNode )treeNode {`
`35`	`39`	`NSLog(@"Node %@ selected", treeNode.title);`
`36`	`40`	`}`