diff --git a/CM20315_Deep2.ipynb b/CM20315_Deep2.ipynb
index 81cb9297..b6b9bff1 100644
--- a/CM20315_Deep2.ipynb
+++ b/CM20315_Deep2.ipynb
@@ -5,7 +5,7 @@
     "colab": {
       "provenance": [],
       "collapsed_sections": [],
-      "authorship_tag": "ABX9TyO4Wn0t5/zNQK7Mjh6ATlij",
+      "authorship_tag": "ABX9TyP87B9tfgXpVQdlQBUGw4mg",
       "include_colab_link": true
     },
     "kernelspec": {
@@ -272,7 +272,7 @@
       "cell_type": "code",
       "source": [
         "# define sizes\n",
-        "D_i=4; D_1=5; D_2=2; D_3=1; D_o=1\n",
+        "D_i=4; D_1=5; D_2=2; D_3=4; D_o=1\n",
         "# We'll choose the inputs and parameters of this network randomly using np.random.normal\n",
         "# For example, we'll set the input using\n",
         "n_data = 10;\n",