diff --git a/CM20315_Shallow.ipynb b/CM20315_Shallow.ipynb
index f381e27..049afd3 100644
--- a/CM20315_Shallow.ipynb
+++ b/CM20315_Shallow.ipynb
@@ -569,11 +569,11 @@
     {
       "cell_type": "code",
       "source": [
-        "# Define a shallow neural network with, one input, one output, and three hidden units\n",
+        "# Define a shallow neural network with, two inputs, one output, and three hidden units\n",
         "def shallow_2_1_3(x1,x2, activation_fn, phi_0,phi_1,phi_2,phi_3, theta_10, theta_11,\\\n",
         "                  theta_12, theta_20, theta_21, theta_22, theta_30, theta_31, theta_32):\n",
-        "  # TODO Replace the lines below to compute the three initial lines \n",
-        "  # (figure 3.3a-c) from the theta parameters.  These are the preactivations\n",
+        "  # TODO Replace the lines below to compute the three initial linear functions \n",
+        "  # (figure 3.8a-c) from the theta parameters.  These are the preactivations\n",
         "  pre_1 = np.zeros_like(x1)\n",
         "  pre_2 = np.zeros_like(x1)\n",
         "  pre_3 = np.zeros_like(x1)\n",
@@ -626,4 +626,4 @@
       "outputs": []
     }
   ]
-}
\ No newline at end of file
+}