diff --git a/dashboard/Sugartrail.ipynb b/dashboard/Sugartrail.ipynb
index 2f98ff6..fcfbc39 100644
--- a/dashboard/Sugartrail.ipynb
+++ b/dashboard/Sugartrail.ipynb
@@ -16,7 +16,7 @@
    "outputs": [],
    "source": [
     "from sugartrail import mapview, api, base\n",
-    "import pandas as pd\n",
+    "# import pandas as pd\n",
     "import ipywidgets as widgets\n",
     "from IPython.display import display, HTML\n",
     "import requests"
@@ -126,12 +126,12 @@
     "\n",
     "def init_network():\n",
     "    init_button.disabled=True\n",
-    "#     api.basic_auth.username = API_input.value\n",
-    "    response = api.get_company(str(company_text.value))\n",
+    "    network.company_id = str(company_text.value)\n",
+    "    response = api.get_company(network.company_id)\n",
     "    company_text.disabled=True\n",
     "    if response:\n",
     "#         network.company_id = str(company_text.value)\n",
-    "        network = base.Network(company=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",