diff --git a/extra_capsnets.ipynb b/extra_capsnets.ipynb
index 08c9433..cbe6086 100644
--- a/extra_capsnets.ipynb
+++ b/extra_capsnets.ipynb
@@ -493,14 +493,14 @@
     "$\n",
     "\\pmatrix{\n",
     "\\mathbf{A} & \\mathbf{B} & \\mathbf{C} \\\\\n",
-    "\\mathbf{D} & \\mathbf{E} & \\mathbf{F} \\\\\n",
+    "\\mathbf{D} & \\mathbf{E} & \\mathbf{F}\n",
     "} \\times\n",
     "\\pmatrix{\n",
     "\\mathbf{G} & \\mathbf{H} & \\mathbf{I} \\\\\n",
-    "\\mathbf{J} & \\mathbf{K} & \\mathbf{L} \\\\\n",
+    "\\mathbf{J} & \\mathbf{K} & \\mathbf{L}\n",
     "} = \\pmatrix{\n",
     "\\mathbf{AG} & \\mathbf{BH} & \\mathbf{CI} \\\\\n",
-    "\\mathbf{DJ} & \\mathbf{EK} & \\mathbf{FL} \\\\\n",
+    "\\mathbf{DJ} & \\mathbf{EK} & \\mathbf{FL}\n",
     "}\n",
     "$"
    ]
@@ -512,16 +512,16 @@
     "We can apply this function to compute $\\hat{\\mathbf{u}}_{j|i}$ for every pair of capsules ($i$, $j$) like this (recall that there are 6×6×32=1152 capsules in the first layer, and 10 in the second layer):\n",
     "$\n",
     "\\pmatrix{\n",
-    "\\mathbf{W}_{1,1} & \\mathbf{W}_{2,1} & \\cdots & \\mathbf{W}_{10,1} \\\\\n",
-    "\\mathbf{W}_{1,2} & \\mathbf{W}_{2,2} & \\cdots & \\mathbf{W}_{10,2} \\\\\n",
-    "\\vdots & \\vdots & \\ddots & \\vdots \\\\\n",
-    "\\mathbf{W}_{1,1152} & \\mathbf{W}_{2,1152} & \\cdots & \\mathbf{W}_{10,1152} \\\\\n",
+    "  \\mathbf{W}_{1,1} & \\mathbf{W}_{2,1} & \\cdots & \\mathbf{W}_{10,1} \\\\\n",
+    "  \\mathbf{W}_{1,2} & \\mathbf{W}_{2,2} & \\cdots & \\mathbf{W}_{10,2} \\\\\n",
+    "  \\vdots & \\vdots & \\ddots & \\vdots \\\\\n",
+    "  \\mathbf{W}_{1,1152} & \\mathbf{W}_{2,1152} & \\cdots & \\mathbf{W}_{10,1152}\n",
     "} \\times\n",
     "\\pmatrix{\n",
-    "\\mathbf{u}_1 & \\mathbf{u}_1 & \\cdots & \\mathbf{u}_1 \\\\\n",
-    "\\mathbf{u}_2 & \\mathbf{u}_2 & \\cdots & \\mathbf{u}_2 \\\\\n",
-    "\\vdots & \\vdots & \\ddots & \\vdots \\\\\n",
-    "\\mathbf{u}_{1152} & \\mathbf{u}_{1152} & \\cdots & \\mathbf{u}_{1152} \\\\\n",
+    "  \\mathbf{u}_1 & \\mathbf{u}_1 & \\cdots & \\mathbf{u}_1 \\\\\n",
+    "  \\mathbf{u}_2 & \\mathbf{u}_2 & \\cdots & \\mathbf{u}_2 \\\\\n",
+    "  \\vdots & \\vdots & \\ddots & \\vdots \\\\\n",
+    "  \\mathbf{u}_{1152} & \\mathbf{u}_{1152} & \\cdots & \\mathbf{u}_{1152}\n",
     "}\n",
     "=\n",
     "\\pmatrix{\n",