diff --git a/Sources/SimilaritySearchKit/Core/Index/SimilarityIndex.swift b/Sources/SimilaritySearchKit/Core/Index/SimilarityIndex.swift
index 6e85b4d..cb1559e 100644
--- a/Sources/SimilaritySearchKit/Core/Index/SimilarityIndex.swift
+++ b/Sources/SimilaritySearchKit/Core/Index/SimilarityIndex.swift
@@ -245,11 +245,12 @@ extension SimilarityIndex {
         }
     }
 
-    public func addItems(_ items: [IndexItem]) {
+    public func addItems(_ items: [IndexItem], completion: (() -> Void)? = nil) {
         Task {
             for item in items {
                 await self.addItem(id: item.id, text: item.text, metadata: item.metadata, embedding: item.embedding)
             }
+            completion?()
         }
     }
 
@@ -326,8 +327,9 @@ extension SimilarityIndex {
     public func loadIndex(fromDirectory path: URL? = nil, name: String? = nil) throws -> [IndexItem]? {
         if let indexPath = try getIndexPath(fromDirectory: path, name: name) {
             let loadedIndexItems = try vectorStore.loadIndex(from: indexPath)
-            addItems(loadedIndexItems)
-            print("Loaded \(indexItems.count) index items from \(indexPath.absoluteString)")
+            addItems(loadedIndexItems) {[self] in
+                print("Loaded \(indexItems.count) index items from \(indexPath.absoluteString)")
+            }
             return loadedIndexItems
         }