diff --git a/requirements.txt b/requirements.txt
index f67c8af..6bba818 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,5 @@
 fastapi
 uvicorn
 debugpy
-langchain_community
\ No newline at end of file
+langchain_community
+langchain
\ No newline at end of file
diff --git a/run.sh b/run.sh
index fccefce..a17de61 100644
--- a/run.sh
+++ b/run.sh
@@ -1,5 +1,5 @@
 python -m venv venv
 source venv/Scripts/activate
 pip install -r requirements.txt
-python signal-processing.py
+python src/test.py
 
diff --git a/src/categorize_expenses.py b/src/categorize_expenses.py
deleted file mode 100644
index a6809ed..0000000
--- a/src/categorize_expenses.py
+++ /dev/null
@@ -1,4 +0,0 @@
-from langchain_community.llms import Ollama
-
-llm = Ollama(model="llama2-uncensored")
-print(llm.invoke("was neil armstrong gay"))
diff --git a/src/test.py b/src/test.py
new file mode 100644
index 0000000..ad8f20f
--- /dev/null
+++ b/src/test.py
@@ -0,0 +1,8 @@
+from langchain.llms import Ollama
+from langchain.callbacks.manager import CallbackManager
+from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler 
+                                 
+llm = Ollama(model="llama2-uncensored", 
+             callback_manager = CallbackManager([StreamingStdOutCallbackHandler()]))
+
+llm("Tell me 5 facts about Roman history:")
\ No newline at end of file