1 files changed, 8 insertions, 5 deletions
diff --git a/search/set.go b/search/set.go
index ed5b442..58ac2fb 100644
--- a/search/set.go
+++ b/search/set.go
@@ -1,16 +1,19 @@
 package search
 
-import "fmt"
+import (
+	"fmt"
+	"codeberg.org/snonux/algorithms/ds"
+)
 
 var (
 	NotFound       = fmt.Errorf("could not find entry")
 	NotImplemented = fmt.Errorf("method not implemented")
 )
 
-type Set interface {
+type Set[K ds.Integer,V ds.Number] interface {
 	Empty() bool
 	Size() int
-	Put(key int, val int)
-	Get(key int) (int, error)
-	Del(key int) (int, error)
+	Put(key K, val V)
+	Get(key K) (V, error)
+	Del(key K) (V, error)
 }