diff --git a/demo.ipynb b/demo.ipynb
index ce6ba86..5f88d0e 100644
--- a/demo.ipynb
+++ b/demo.ipynb
@@ -85,11 +85,11 @@
     "# Export the proper environment variables based on the client you are using.\n",
     "\n",
     "# OpenAI API Key\n",
-    "# os.environ[\"OPENAI_ORGANIZATION\"] = OpenAICostTrackerUtils.get_api_key(path='openai_organization.txt')\n",
-    "os.environ[\"OPENAI_API_KEY\"] = OpenAICostTrackerUtils.get_api_key(path='/Users/mikolajboronski/.openai')\n",
+    "os.environ[\"OPENAI_ORGANIZATION\"] = OpenAICostTrackerUtils.get_api_key(path='openai_organization.txt')\n",
+    "os.environ[\"OPENAI_API_KEY\"] = OpenAICostTrackerUtils.get_api_key(path='openai_api_key.txt')\n",
     "\n",
     "# Azure OpenAI API Key\n",
-    "# os.environ[\"AZURE_OPENAI_KEY\"] = OpenAICostTrackerUtils.get_api_key('azure_openai_key.txt')"
+    "os.environ[\"AZURE_OPENAI_KEY\"] = OpenAICostTrackerUtils.get_api_key('azure_openai_key.txt')"
    ]
   },
   {
@@ -105,7 +105,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "experiment_name = \"Demo2\"\n",
+    "experiment_name = \"Demo\"\n",
     "messages = [\n",
     "    {\"role\": \"system\", \"content\": \"You are a helpful assistant.\"},\n",
     "    {\"role\": \"user\", \"content\": \"Who won the euro 2020?\"},\n",