diff --git a/1-ollama-langchain/main.py b/1-ollama-langchain/main.py
index 7dc0922..cac0cfb 100644
--- a/1-ollama-langchain/main.py
+++ b/1-ollama-langchain/main.py
@@ -1,9 +1,9 @@
 from langchain.callbacks.manager import CallbackManager
 from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
-from langchain.llms import Ollama
+from langchain_community.llms import Ollama
 
 llm = Ollama(
     model="mistral", callback_manager=CallbackManager([StreamingStdOutCallbackHandler()])
 )
 
-print(llm("Tell me about the history of AI"))
\ No newline at end of file
+print(llm.invoke("Tell me a joke"))
\ No newline at end of file
diff --git a/app.ipynb b/app.ipynb
new file mode 100644
index 0000000..e69de29