diff --git a/ch06/01_main-chapter-code/ch06.ipynb b/ch06/01_main-chapter-code/ch06.ipynb
index 44cff18..dda54d9 100644
--- a/ch06/01_main-chapter-code/ch06.ipynb
+++ b/ch06/01_main-chapter-code/ch06.ipynb
@@ -1513,7 +1513,7 @@
    "id": "669e1fd1-ace8-44b4-b438-185ed0ba8b33",
    "metadata": {},
    "source": [
-    "<img src=\"https://sebastianraschka.com/images/LLMs-from-scratch-images/ch06_compressed/overview-3.webp\" width=500px>"
+    "<img src=\"https://sebastianraschka.com/images/LLMs-from-scratch-images/ch06_compressed/overview-3.webp?123\" width=500px>"
    ]
   },
   {
@@ -1524,6 +1524,14 @@
     "- Before explaining the loss calculation, let's have a brief look at how the model outputs are turned into class labels"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "557996dd-4c6b-49c4-ab83-f60ef7e1d69e",
+   "metadata": {},
+   "source": [
+    "<img src=\"https://sebastianraschka.com/images/LLMs-from-scratch-images/ch06_compressed/class-argmax.webp\" width=600px>"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 26,
@@ -2347,7 +2355,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.4"
+   "version": "3.10.6"
   }
  },
  "nbformat": 4,