diff --git a/HW 5/Iris_knn_W2025_template.ipynb b/HW 5/Iris_knn_W2025_template.ipynb
index 3423242..8f8ef94 100644
--- a/HW 5/Iris_knn_W2025_template.ipynb	
+++ b/HW 5/Iris_knn_W2025_template.ipynb	
@@ -2700,7 +2700,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 35,
+      "execution_count": null,
       "metadata": {
         "id": "AsC4oa2qKPSG"
       },
@@ -2722,7 +2722,7 @@
         "# final predictive model (k-nearest-neighbor), with tuned k + ALL data incorporated\n",
         "#\n",
         "\n",
-        "def predictiveModel( features: list[float] ) -> None:\n",
+        "def predictiveModel( features: list[float] ) -> str:\n",
         "    ''' input: a list of four features\n",
         "                [ sepallen, sepalwid, petallen, petalwid ]\n",
         "        output: the predicted species of iris, from\n",