diff --git a/learn_the_basics/03_compile_code.ipynb b/learn_the_basics/03_compile_code.ipynb
index 49077084..611fccc7 100644
--- a/learn_the_basics/03_compile_code.ipynb
+++ b/learn_the_basics/03_compile_code.ipynb
@@ -32,7 +32,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install ivy"
+    "!pip install ivy\n",
+    "!pip install torch\n",
+    "!pip install jax\n",
+    "!pip intall jaxlib"
    ]
   },
   {
@@ -55,19 +58,10 @@
     "def normalize(x):\n",
     "    mean = torch.mean(x)\n",
     "    std = torch.std(x)\n",
-    "    return torch.div(torch.sub(x, mean), std)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "outputs": [],
-   "source": [
+    "    return torch.div(torch.sub(x, mean), std)\n",
+    "\n",
     "normalize = ivy.unify(normalize, source=\"torch\")"
-   ],
-   "metadata": {
-    "collapsed": false
-   }
+   ]
   },
   {
    "attachments": {},