more on tree

1 files changed, 13 insertions, 4 deletions

diff --git a/set/tree.go b/set/tree.go
index d4f7b7d..8233b0e 100644
--- a/set/tree.go
+++ b/set/tree.go
@@ -66,7 +66,16 @@ func (t *Tree) Get(key int) (int, error) {
 	return t.get(t.root, key)
 }
 
-/*
-	Get(key int) (int, error)
-	Del(key int) (int, error)
-*/
+func (t *Tree) Del(key int) (int, error) {
+	if t.root == nil {
+		return 0, NotFound
+	}
+
+	if t.root.key == key {
+		val := t.root.val
+
+		return val, nil
+	}
+
+	return 0, NotImplemented
+}