diff --git a/spinups/spin_up_the_simplest.ipynb b/spinups/spin_up_the_simplest.ipynb
index d11d570..9f3d16a 100644
--- a/spinups/spin_up_the_simplest.ipynb
+++ b/spinups/spin_up_the_simplest.ipynb
@@ -6,47 +6,14 @@
       "metadata": {},
       "outputs": [],
       "source": [
-        "!git clone https://github.com/anas-rz/k3im.git"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": 1,
-      "metadata": {},
-      "outputs": [],
-      "source": [
-        "import os\n",
-        "os.environ['KERAS_BACKEND'] = 'torch'"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": 2,
-      "metadata": {
-        "id": "NOZYc1_yDiEK"
-      },
-      "outputs": [
-        {
-          "name": "stdout",
-          "output_type": "stream",
-          "text": [
-            "/bin/bash: /home/anas/miniconda3/envs/keras_core/lib/libtinfo.so.6: no version information available (required by /bin/bash)\n",
-            "Keras version: 3.0.0\n"
-          ]
-        }
-      ],
-      "source": [
-        "!pip install keras -qq --upgrade"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": 3,
-      "metadata": {},
-      "outputs": [],
-      "source": [
+        "!pip install keras medmnist -qq --upgrade\n",
+        "!git clone https://github.com/anas-rz/k3im.git\n",
+        "\n",
         "import sys\n",
-        "sys.path.append('/content/k3im')"
+        "import os\n",
+        "\n",
+        "sys.path.append('/content/k3im')\n",
+        "os.environ['KERAS_BACKEND'] = 'jax'"
       ]
     },
     {
diff --git a/spinups/spin_up_the_simplest_1d.ipynb b/spinups/spin_up_the_simplest_1d.ipynb
index e21dece..a9de369 100644
--- a/spinups/spin_up_the_simplest_1d.ipynb
+++ b/spinups/spin_up_the_simplest_1d.ipynb
@@ -6,75 +6,16 @@
       "metadata": {},
       "outputs": [],
       "source": [
-        "!git clone https://github.com/anas-rz/k3im.git"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": 1,
-      "metadata": {},
-      "outputs": [],
-      "source": [
+        "!pip install keras --upgrade\n",
+        "!git clone https://github.com/anas-rz/k3im.git\n",
+        "\n",
+        "import sys\n",
         "import os\n",
+        "\n",
+        "sys.path.append('/content/k3im')\n",
         "os.environ['KERAS_BACKEND'] = 'jax'"
       ]
     },
-    {
-      "cell_type": "code",
-      "execution_count": 2,
-      "metadata": {
-        "id": "NOZYc1_yDiEK"
-      },
-      "outputs": [
-        {
-          "name": "stdout",
-          "output_type": "stream",
-          "text": [
-            "/bin/bash: /home/anas/miniconda3/envs/keras_core/lib/libtinfo.so.6: no version information available (required by /bin/bash)\n",
-            "Requirement already satisfied: keras in /home/anas/miniconda3/envs/keras_core/lib/python3.11/site-packages (3.0.0)\n",
-            "Requirement already satisfied: absl-py in /home/anas/miniconda3/envs/keras_core/lib/python3.11/site-packages (from keras) (1.4.0)\n",
-            "Requirement already satisfied: numpy in /home/anas/miniconda3/envs/keras_core/lib/python3.11/site-packages (from keras) (1.24.3)\n",
-            "Requirement already satisfied: rich in /home/anas/miniconda3/envs/keras_core/lib/python3.11/site-packages (from keras) (13.4.2)\n",
-            "Requirement already satisfied: namex in /home/anas/miniconda3/envs/keras_core/lib/python3.11/site-packages (from keras) (0.0.7)\n",
-            "Requirement already satisfied: h5py in /home/anas/miniconda3/envs/keras_core/lib/python3.11/site-packages (from keras) (3.9.0)\n",
-            "Requirement already satisfied: dm-tree in /home/anas/miniconda3/envs/keras_core/lib/python3.11/site-packages (from keras) (0.1.8)\n",
-            "Requirement already satisfied: markdown-it-py>=2.2.0 in /home/anas/miniconda3/envs/keras_core/lib/python3.11/site-packages (from rich->keras) (3.0.0)\n",
-            "Requirement already satisfied: pygments<3.0.0,>=2.13.0 in /home/anas/miniconda3/envs/keras_core/lib/python3.11/site-packages (from rich->keras) (2.15.1)\n",
-            "Requirement already satisfied: mdurl~=0.1 in /home/anas/miniconda3/envs/keras_core/lib/python3.11/site-packages (from markdown-it-py>=2.2.0->rich->keras) (0.1.2)\n"
-          ]
-        }
-      ],
-      "source": [
-        "!pip install keras --upgrade"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": 3,
-      "metadata": {},
-      "outputs": [],
-      "source": [
-        "import sys\n",
-        "sys.path.append('/content/k3im')"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": 4,
-      "metadata": {},
-      "outputs": [
-        {
-          "name": "stdout",
-          "output_type": "stream",
-          "text": [
-            "Keras version: 3.0.0\n"
-          ]
-        }
-      ],
-      "source": [
-        "check_keras_3()"
-      ]
-    },
     {
       "cell_type": "code",
       "execution_count": 5,
diff --git a/spinups/spin_up_the_simplest_3d.ipynb b/spinups/spin_up_the_simplest_3d.ipynb
index 5b4070d..0170014 100644
--- a/spinups/spin_up_the_simplest_3d.ipynb
+++ b/spinups/spin_up_the_simplest_3d.ipynb
@@ -6,49 +6,16 @@
       "metadata": {},
       "outputs": [],
       "source": [
-        "!git clone https://github.com/anas-rz/k3im.git"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": 1,
-      "metadata": {},
-      "outputs": [],
-      "source": [
+        "!pip install keras medmnist -qq --upgrade\n",
+        "!git clone https://github.com/anas-rz/k3im.git\n",
+        "\n",
+        "import sys\n",
         "import os\n",
+        "\n",
+        "sys.path.append('/content/k3im')\n",
         "os.environ['KERAS_BACKEND'] = 'jax'"
       ]
     },
-    {
-      "cell_type": "code",
-      "execution_count": 2,
-      "metadata": {
-        "id": "NOZYc1_yDiEK"
-      },
-      "outputs": [
-        {
-          "name": "stdout",
-          "output_type": "stream",
-          "text": [
-            "/bin/bash: /home/anas/miniconda3/envs/keras_core/lib/libtinfo.so.6: no version information available (required by /bin/bash)\n",
-            "Keras version: 3.0.0\n"
-          ]
-        }
-      ],
-      "source": [
-        "!pip install keras medmnist -qq --upgrade"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": 3,
-      "metadata": {},
-      "outputs": [],
-      "source": [
-        "import sys\n",
-        "sys.path.append('/content/k3im')"
-      ]
-    },
     {
       "cell_type": "code",
       "execution_count": 4,
diff --git a/spinups/spin_up_the_simplest_video.ipynb b/spinups/spin_up_the_simplest_video.ipynb
index 92371a9..9636bb2 100644
--- a/spinups/spin_up_the_simplest_video.ipynb
+++ b/spinups/spin_up_the_simplest_video.ipynb
@@ -6,43 +6,15 @@
       "metadata": {},
       "outputs": [],
       "source": [
-        "!git clone https://github.com/anas-rz/k3im.git"
+        "!pip install keras medmnist -qq --upgrade\n",
+        "!git clone https://github.com/anas-rz/k3im.git\n",
+        "\n",
+        "import sys\n",
+        "import os\n",
+        "\n",
+        "sys.path.append('/content/k3im')\n",
+        "os.environ['KERAS_BACKEND'] = 'jax'"
       ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": 1,
-      "metadata": {
-        "id": "NOZYc1_yDiEK"
-      },
-      "outputs": [
-        {
-          "name": "stdout",
-          "output_type": "stream",
-          "text": [
-            "Requirement already satisfied: keras in /home/anas/miniconda3/envs/dfd_keras_core/lib/python3.11/site-packages (3.0.0)\n",
-            "Requirement already satisfied: absl-py in /home/anas/miniconda3/envs/dfd_keras_core/lib/python3.11/site-packages (from keras) (2.0.0)\n",
-            "Requirement already satisfied: numpy in /home/anas/miniconda3/envs/dfd_keras_core/lib/python3.11/site-packages (from keras) (1.24.3)\n",
-            "Requirement already satisfied: rich in /home/anas/miniconda3/envs/dfd_keras_core/lib/python3.11/site-packages (from keras) (13.7.0)\n",
-            "Requirement already satisfied: namex in /home/anas/miniconda3/envs/dfd_keras_core/lib/python3.11/site-packages (from keras) (0.0.7)\n",
-            "Requirement already satisfied: h5py in /home/anas/miniconda3/envs/dfd_keras_core/lib/python3.11/site-packages (from keras) (3.9.0)\n",
-            "Requirement already satisfied: dm-tree in /home/anas/miniconda3/envs/dfd_keras_core/lib/python3.11/site-packages (from keras) (0.1.8)\n",
-            "Requirement already satisfied: markdown-it-py>=2.2.0 in /home/anas/miniconda3/envs/dfd_keras_core/lib/python3.11/site-packages (from rich->keras) (3.0.0)\n",
-            "Requirement already satisfied: pygments<3.0.0,>=2.13.0 in /home/anas/miniconda3/envs/dfd_keras_core/lib/python3.11/site-packages (from rich->keras) (2.17.2)\n",
-            "Requirement already satisfied: mdurl~=0.1 in /home/anas/miniconda3/envs/dfd_keras_core/lib/python3.11/site-packages (from markdown-it-py>=2.2.0->rich->keras) (0.1.2)\n"
-          ]
-        }
-      ],
-      "source": [
-        "!pip install keras --upgrade"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {},
-      "outputs": [],
-      "source": []
     }
   ],
   "metadata": {