diff --git a/heap.go b/heap.go
index f4d5b70..0c7789e 100644
--- a/heap.go
+++ b/heap.go
@@ -2,7 +2,7 @@ package bitknn
 
 import "unsafe"
 
-// neighborHeap is a max-heap that stores distances and their corresponding indices.
+// neighborHeap is a max-heap that stores data point's distances together with their indices in the training set.
 // The heap is used to keep track of nearest neighbors.
 type neighborHeap struct {
 	distances    []int