diff --git a/src/key_search.cpp b/src/key_search.cpp
index 12a3a11..82b0cb5 100644
--- a/src/key_search.cpp
+++ b/src/key_search.cpp
@@ -43,6 +43,11 @@ void KeySearch::addList(const strlist_t& list)
 	}
 }
 
+void KeySearch::addDictionary(const Dictionary& dictionary)
+{
+	m_dict = dictionary;
+}
+
 const strlist_t& KeySearch::getList()
 {
 	return m_words;
diff --git a/src/key_search.h b/src/key_search.h
index f2810fb..da26772 100644
--- a/src/key_search.h
+++ b/src/key_search.h
@@ -24,6 +24,7 @@
 #ifndef EOSIOKEYGEN_KEY_SEARCH_H
 #define EOSIOKEYGEN_KEY_SEARCH_H
 
+#include "core/dictionary.h"
 #include "string.h"
 
 namespace eoskeygen {
@@ -37,6 +38,8 @@ public :
 	// Add a list of words to search for.
 	void addList(const strlist_t& list);
 
+	void addDictionary(const Dictionary& dictionary);
+
 	// get the list of words to search for.
 	const strlist_t& getList();
 
@@ -65,6 +68,9 @@ protected :
 	// List of words to search for.
 	strlist_t m_words;
 
+	// Dictionary to use when we find a search result.
+	Dictionary m_dict;
+
 #ifdef HAVE_THREADS
 	// Number of threads to use.
 	size_t m_threads;