diff --git a/Phase 2/task1.ipynb b/Phase 2/task1.ipynb
index e1063a0..90c6840 100644
--- a/Phase 2/task1.ipynb	
+++ b/Phase 2/task1.ipynb	
@@ -151,6 +151,9 @@
     "  fig, axes = plt.subplots(1, k, figsize=(15, 5))\n",
     "\n",
     "  for i in range(k):\n",
+    "    # caltechDataset[similar_images[i][\"_id\"]][1] because\n",
+    "    # similar_images[i][\"_id\"] will provide me the image id\n",
+    "    # [1] will be image pixel values since caltechDataset is in format (id, pixels, label)\n",
     "    axes[i].imshow(caltechDataset[similar_images[i][\"_id\"]][1].permute(1, 2, 0))\n",
     "    axes[i].set_title(f'id: {similar_images[i][\"_id\"]}')\n",
     "\n",