diff --git a/notebooks/sota_hackathon.ipynb b/notebooks/sota_hackathon.ipynb
index fd6e59d4..3500a4ae 100644
--- a/notebooks/sota_hackathon.ipynb
+++ b/notebooks/sota_hackathon.ipynb
@@ -138,10 +138,7 @@
     "    \"trajectory_dir\": str(Path(\"trajectories\")),\n",
     "    \"only_n_most_recent_images\": 3,\n",
     "    \"verbosity\": logging.INFO\n",
-    "}\n",
-    "\n",
-    "# Create agent\n",
-    "agent = ComputerAgent(**agent_config)"
+    "}"
    ]
   },
   {
@@ -172,7 +169,7 @@
    "outputs": [],
    "source": [
     "from computer import Computer, VMProviderType\n",
-    "import webbrowser  \n",
+    "import webbrowser\n",
     "\n",
     "# Connect to your existing cloud container\n",
     "computer = Computer(\n",
@@ -182,7 +179,7 @@
     ")\n",
     "await computer.run()\n",
     "\n",
-    "agent_config[\"tools\"] = computer\n",
+    "agent_config[\"tools\"] = [ computer ]\n",
     "\n",
     "webbrowser.open(\"http://localhost:8006/\", new=0, autoraise=True)"
    ]
@@ -204,8 +201,11 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "# Create agent\n",
+    "agent = ComputerAgent(**agent_config)\n",
+    "\n",
     "tasks = [\n",
-    "    \"Look for a repository named trycua/cua on GitHub.\"\n",
+    "    \"Open the web browser and search for a repository named trycua/cua on GitHub.\"\n",
     "]\n",
     "\n",
     "for i, task in enumerate(tasks):\n",