diff --git a/Week 2 Programming Assignment.ipynb b/Week 2 Programming Assignment.ipynb
index edfe8fa..c4d776d 100644
--- a/Week 2 Programming Assignment.ipynb	
+++ b/Week 2 Programming Assignment.ipynb	
@@ -179,14 +179,14 @@
     "    The seed argument should be passed to the flow_from_directory method.\n",
     "    \n",
     "    \"\"\"\n",
-    " # I couldn't get this one right. If you get it let me know So, I can recitfy.\n",
+    " 
     "    image_data_gen = image_data_generator.flow_from_directory(\n",
     "        directory = directory,\n",
+    "        classes=['classroom','conference_room','church_outdoor'],\n",
     "        batch_size = 20,\n",
     "        target_size = (64,64),\n",
-    "        \n",
+    "        seed=seed,\n",
     "        class_mode = \"categorical\"\n",
-    "        \n",
     "        )\n",
     "    return image_data_gen\n",
     "    "