diff --git a/source/oci_genai_llm_context.py b/source/oci_genai_llm_context.py
index edef52b..890ae84 100644
--- a/source/oci_genai_llm_context.py
+++ b/source/oci_genai_llm_context.py
@@ -1,4 +1,3 @@
-import chromadb
 from langchain_community.document_loaders import PyPDFLoader
 from langchain_community.chat_models.oci_generative_ai import ChatOCIGenAI
 from langchain_core.prompts import PromptTemplate
@@ -46,6 +45,8 @@ def chat():
     )
     retriever = vectorstore.as_retriever()
 
+    #### YOU CAN PERSIST YOUR VECTOR STORE HERE
+
     template = """ 
         If the query in question is not a comparison between SOA SUITE and OIC, consider only the documents relevant to the subject,
          that is, if the question is about SOA SUITE, consider only the SOA SUITE documents. If the question is about OIC,