diff --git a/Phase 2/task_10.ipynb b/Phase 2/task_10.ipynb
index 7f06f5a..f5e3cf2 100644
--- a/Phase 2/task_10.ipynb	
+++ b/Phase 2/task_10.ipynb	
@@ -158,6 +158,15 @@
     "      else:\n",
     "        comparison_vector = np.matmul(label_rep, np.transpose(H))\n",
     "\n",
+    "    case \"lda\":\n",
+    "      comparison_feature_space = np.array(data[\"image-semantic\"])\n",
+    "      label_vectors = []\n",
+    "      length = len(comparison_feature_space)\n",
+    "      for i in range(length):\n",
+    "        if all_images[i][\"true_label\"] == label:\n",
+    "          label_vectors.append(comparison_feature_space[i])\n",
+    "      comparison_vector = [sum(col) / len(col) for col in zip(*label_vectors)] \n",
+    "        \n",
     "    case \"kmeans\":\n",
     "      comparison_vector = []\n",
     "      comparison_feature_space = np.array(data[\"image-semantic\"])\n",
@@ -240,6 +249,10 @@
     "      comparison_feature_space = np.array(data['image-semantic'])\n",
     "      comparison_vector = comparison_feature_space[label]\n",
     "\n",
+    "    case \"lda\":\n",
+    "      comparison_feature_space = np.array(data['image-semantic'])\n",
+    "      comparison_vector = comparison_feature_space[label]\n",
+    "\n",
     "    case \"kmeans\":\n",
     "      comparison_feature_space = np.array(data[\"image-semantic\"])\n",
     "      comparison_vector = comparison_feature_space[label]\n",