@@ -82,7 +82,7 @@
    "source": [
     "node_schema = []\n",
     "node_pattern = \"(?P<directory>.+)/(?P<label>.+)\\.(?P<entity>.+)\\.csv\"\n",
-    "for filename in glob(\"csv/*.nodes.csv\"):\n",
+    "for filename in glob(\"csv_jakleen/*.nodes.csv\"):\n",
     "    match = re.match(node_pattern, filename).groupdict()\n",
     "    label = match[\"label\"].replace(\"_\", \" \")\n",
     "    df = pd.read_csv(filename)\n",
@@ -185,7 +185,7 @@
     "edge_schema = []\n",
     "edge_pattern = \"(?P<directory>.+)/(?P<source_type>.+)\\.(?P<relation>.+)\\.(?P<target_type>.+)\\.(?P<entity>.+)\\.csv\"\n",
     "\n",
-    "for filename in glob(\"csv/*.edges.csv\"):\n",
+    "for filename in glob(\"csv_jakleen/*.edges.csv\"):\n",
     "   match = re.match(edge_pattern, filename).groupdict()\n",
     "   entity = match[\"entity\"]\n",
     "   source_type = match[\"source_type\"]\n",
@@ -223,6 +223,26 @@
     "         "
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "2"
+      ]
+     },
+     "execution_count": 4,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "len(edge_schema)"
+   ]
+  },
   {
    "attachments": {},
    "cell_type": "markdown",
@@ -242,11 +262,26 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
-    "title = \"Enrichr-KG\"\n",
+    "# First you need to modify this based on your dataset. \n",
+    "# This will be the initial query that the users will see when visiting your site\n",
+    "initial_query = {\n",
+    "\t\"start\": \"Gene\",\n",
+    "\t\"start_term\": \"APOE\",\n",
+    "\t\"start_field\": \"label\"\n",
+    "}"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "title = \"Demo-KG\"\n",
     "icon = \"https://minio.dev.maayanlab.cloud/enrichr-kg/enrichr-kg.png\"\n",
     "rest = {\n",
     "    \"header\": {\n",
@@ -273,11 +308,7 @@
     "\t\t\t\t\"component\": \"KnowledgeGraph\",\n",
     "\t\t\t\t\"props\": {\n",
     "\t\t\t\t\t\"init_function\": \"initialize_kg\",\n",
-    "\t\t\t\t\t\"initial_query\": {\n",
-    "\t\t\t\t\t\t\"start\": \"Gene\",\n",
-    "\t\t\t\t\t\t\"start_term\": \"APOE\",\n",
-    "\t\t\t\t\t\t\"start_field\": \"label\"\n",
-    "\t\t\t\t\t},\n",
+    "\t\t\t\t\t\"initial_query\": initial_query,\n",
     "\t\t\t\t\t\"subheader\": {\n",
     "\t\t\t\t\t\t\"url_field\": \"filter\",\n",
     "\t\t\t\t\t\t\"query_field\": \"relation\"\n",
@@ -356,7 +387,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 6,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -366,7 +397,7 @@
     "    **rest\n",
     "}\n",
     "\n",
-    "with open(\"../schema.json\", 'w') as o:\n",
+    "with open(\"../schema_sample.json\", 'w') as o:\n",
     "    o.write(json.dumps(schema, indent=4))"
    ]
   },
@@ -480,7 +511,11 @@
     "docker-compose up\n",
     "```\n",
     "you can check http://localhost:3000 again\n",
-    "\n"
+    "\n",
+    "```\n",
+    "# push to dockerhub\n",
+    "docker-compose push kg\n",
+    "```"
    ]
   },
   {