diff --git a/labs/ex01/solution/taskB.ipynb b/labs/ex01/solution/taskB.ipynb
index 8f4f067f2..fa63edfde 100644
--- a/labs/ex01/solution/taskB.ipynb
+++ b/labs/ex01/solution/taskB.ipynb
@@ -83,8 +83,8 @@
     "    A naive solution for finding pairvise distances between poins in P and Q\n",
     "\n",
     "    Args:\n",
-    "        P: numpy array of shape=(p, 2)\n",
-    "        Q: numpy array of shape=(q, 2)\n",
+    "        P: numpy array of shape=(p, n)\n",
+    "        Q: numpy array of shape=(q, n)\n",
     "    Returns:\n",
     "        D: numpy array of shape=(p, q)\n",
     "\n",
@@ -247,8 +247,8 @@
     "    An optimized solution using matching indices\n",
     "\n",
     "    Args:\n",
-    "        P: numpy array of shape=(p, 2)\n",
-    "        Q: numpy array of shape=(q, 2)\n",
+    "        P: numpy array of shape=(p, n)\n",
+    "        Q: numpy array of shape=(q, n)\n",
     "    Returns:\n",
     "        D: numpy array of shape=(p, q)\n",
     "\n",
@@ -284,8 +284,8 @@
     "    An optimized solution using matching indices\n",
     "\n",
     "    Args:\n",
-    "        P: numpy array of shape=(p, 2)\n",
-    "        Q: numpy array of shape=(q, 2)\n",
+    "        P: numpy array of shape=(p, n)\n",
+    "        Q: numpy array of shape=(q, n)\n",
     "    Returns:\n",
     "        D: numpy array of shape=(p, q)\n",
     "\n",
@@ -324,8 +324,8 @@
     "    A solution using scipy\n",
     "\n",
     "    Args:\n",
-    "        P: numpy array of shape=(p, 2)\n",
-    "        Q: numpy array of shape=(q, 2)\n",
+    "        P: numpy array of shape=(p, n)\n",
+    "        Q: numpy array of shape=(q, n)\n",
     "\n",
     "    Returns:\n",
     "        D: numpy array of shape=(p, q)\n",
@@ -354,8 +354,8 @@
     "    A solution using tensor broadcasting\n",
     "\n",
     "    Args:\n",
-    "        P: numpy array of shape=(p, 2)\n",
-    "        Q: numpy array of shape=(q, 2)\n",
+    "        P: numpy array of shape=(p, n)\n",
+    "        Q: numpy array of shape=(q, n)\n",
     "\n",
     "    Returns:\n",
     "        D: numpy array of shape=(p, q)\n",
diff --git a/labs/ex01/template/taskB.ipynb b/labs/ex01/template/taskB.ipynb
index 483ee73d6..30b99ea8f 100644
--- a/labs/ex01/template/taskB.ipynb
+++ b/labs/ex01/template/taskB.ipynb
@@ -60,8 +60,8 @@
     "    A naive solution for finding pairvise distances between poins in P and Q\n",
     "\n",
     "    Args:\n",
-    "        P: numpy array of shape=(p, 2)\n",
-    "        Q: numpy array of shape=(q, 2)\n",
+    "        P: numpy array of shape=(p, n)\n",
+    "        Q: numpy array of shape=(q, n)\n",
     "    Returns:\n",
     "        D: numpy array of shape=(p, q)\n",
     "\n",
@@ -127,8 +127,8 @@
     "    An optimized solution using matching indices\n",
     "\n",
     "    Args:\n",
-    "        P: numpy array of shape=(p, 2)\n",
-    "        Q: numpy array of shape=(q, 2)\n",
+    "        P: numpy array of shape=(p, n)\n",
+    "        Q: numpy array of shape=(q, n)\n",
     "    Returns:\n",
     "        D: numpy array of shape=(p, q)\n",
     "\n",
@@ -176,8 +176,8 @@
     "    A solution using scipy\n",
     "\n",
     "    Args:\n",
-    "        P: numpy array of shape=(p, 2)\n",
-    "        Q: numpy array of shape=(q, 2)\n",
+    "        P: numpy array of shape=(p, n)\n",
+    "        Q: numpy array of shape=(q, n)\n",
     "\n",
     "    Returns:\n",
     "        D: numpy array of shape=(p, q)\n",
@@ -207,8 +207,8 @@
     "    A solution using tensor broadcasting\n",
     "\n",
     "    Args:\n",
-    "        P: numpy array of shape=(p, 2)\n",
-    "        Q: numpy array of shape=(q, 2)\n",
+    "        P: numpy array of shape=(p, n)\n",
+    "        Q: numpy array of shape=(q, n)\n",
     "\n",
     "    Returns:\n",
     "        D: numpy array of shape=(p, q)\n",