diff --git a/_notebooks/2023-08-02-Mario-Tube.ipynb b/_notebooks/2023-08-02-Mario-Tube.ipynb
index 7b7191e..37e18f7 100644
--- a/_notebooks/2023-08-02-Mario-Tube.ipynb
+++ b/_notebooks/2023-08-02-Mario-Tube.ipynb
@@ -22,7 +22,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 15,
+   "execution_count": 16,
    "metadata": {},
    "outputs": [
     {
@@ -151,12 +151,14 @@
        "                player.position.x <= tube.position.x + tube.width\n",
        "            ) {\n",
        "                player.velocity.y = 0;\n",
-       "                gravity = 0.02\n",
        "                player.position.y += 0.1\n",
        "                player.velocity.y = 0.0001\n",
-       "            }\n",
-       "            else {\n",
+       "\n",
+       "                while (player.position.y + player.height < tube.position.y + tube.height) {\n",
+       "                    gravity = 0.02\n",
+       "                }\n",
        "                gravity = 1.5\n",
+       "                \n",
        "            }\n",
        "        \n",
        "        if (\n",
@@ -361,12 +363,14 @@
     "                player.position.x <= tube.position.x + tube.width\n",
     "            ) {\n",
     "                player.velocity.y = 0;\n",
-    "                gravity = 0.02\n",
     "                player.position.y += 0.1\n",
     "                player.velocity.y = 0.0001\n",
-    "            }\n",
-    "            else {\n",
+    "\n",
+    "                while (player.position.y + player.height < tube.position.y + tube.height) {\n",
+    "                    gravity = 0.02\n",
+    "                }\n",
     "                gravity = 1.5\n",
+    "                \n",
     "            }\n",
     "        \n",
     "        if (\n",