diff --git a/dashboard/Sugartrail.ipynb b/dashboard/Sugartrail.ipynb
index f0b5ee0..613fbf2 100644
--- a/dashboard/Sugartrail.ipynb
+++ b/dashboard/Sugartrail.ipynb
@@ -125,15 +125,16 @@
     "generate_network_button.on_click(lambda bt: generate_network()) \n",
     "\n",
     "def init_network():\n",
-    "    api.basic_auth.username = API_input.value\n",
+    "    init_button.disabled=True\n",
+    "    company_text.disabled=True\n",
+    "#     api.basic_auth.username = API_input.value\n",
     "    response = api.get_company(str(company_text.value))\n",
     "    if response:\n",
-    "        network.company_id = str(company_text.value)\n",
+    "#         network.company_id = str(company_text.value)\n",
+    "        network = base.Network(company=str(company_text.value))\n",
     "        init_status.value = u'\\u2705: Initialisation successful for ' + str(response['company_name']) \n",
     "        depth_selector.disabled = False\n",
     "        generate_network_button.disabled = False\n",
-    "        init_button.disabled=True\n",
-    "        company_text.disabled=True\n",
     "    else:\n",
     "        init_button.disabled=False\n",
     "        company_text.disabled=False\n",