diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..c332cc1
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,12 @@
+GRAPHDATA
+GRAPHDATA_OUTPUT
+
+lib
+build
+models
+
+.venv
+.vscode
+.cache
+
+GraphReasoning.egg-info
diff --git a/Notebooks/GraphReasoning - Graph Analysis.ipynb b/Notebooks/GraphReasoning - Graph Analysis.ipynb
index e766113..1f28463 100644
--- a/Notebooks/GraphReasoning - Graph Analysis.ipynb	
+++ b/Notebooks/GraphReasoning - Graph Analysis.ipynb	
@@ -65,7 +65,7 @@
    "source": [
     "#Hugging Face repo\n",
     "repository_id = \"lamm-mit/GraphReasoning\"\n",
-    "data_dir='./GRAPHDATA/'    \n",
+    "data_dir='./GRAPHDATA'    \n",
     "\n",
     "data_dir_output='./GRAPHDATA_OUTPUT/'\n",
     "\n",
diff --git a/setup.py b/setup.py
index aadb365..a8c0864 100644
--- a/setup.py
+++ b/setup.py
@@ -27,10 +27,12 @@
         'torchaudio',
         'huggingface_hub',
         'langchain',
+        'langchain-community',
         'pyvis',
         'yachalk',
         'pytesseract',
         'llama-index',
+        'llama-index-embeddings-huggingface',
         'tqdm',
         'ipython',
         'scikit-learn',
@@ -38,10 +40,10 @@
         'seaborn',
         'uuid',
         'pdfminer.six',
-        'community',
         'guidance',
         'python-louvain',
-        'wkhtmltopdf'
+        'wkhtmltopdf',
+        'llama-cpp-python',
     ],
     description='GraphReasoning: Use LLM to reason over graphs, combined with multi-agent modeling.',
     long_description=long_description,