diff --git a/notebooks/test-service.ipynb b/notebooks/test-service.ipynb
index 6c67d2b70b07f087aa56194c3b788a5f7db64b56..7d3a09c26349aa264a342458a5afe0e58f504e95 100644
--- a/notebooks/test-service.ipynb
+++ b/notebooks/test-service.ipynb
@@ -218,8 +218,7 @@
     {
      "data": {
       "text/plain": [
-       "[b'flatland-rl::FLATLAND_RL_SERVICE_ID::commands',\n",
-       " b'flatland-rl::FLATLAND_RL_SERVICE_ID::response::9233d209716f4ae78a5dbe124de67e27']"
+       "[b'flatland-rl::FLATLAND_RL_SERVICE_ID::response::9233d209716f4ae78a5dbe124de67e27']"
       ]
      },
      "execution_count": 11,
@@ -241,7 +240,6 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Deleting: b'flatland-rl::FLATLAND_RL_SERVICE_ID::commands'\n",
       "Deleting: b'flatland-rl::FLATLAND_RL_SERVICE_ID::response::9233d209716f4ae78a5dbe124de67e27'\n"
      ]
     }
@@ -256,7 +254,16 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "### Remove `/tm"
+    "### Remove `/tmp/output.csv`"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 13,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!rm -f /tmp/output.csv"
    ]
   },
   {
@@ -268,7 +275,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": 14,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -277,7 +284,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": 15,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -293,7 +300,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 15,
+   "execution_count": 16,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -329,7 +336,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 16,
+   "execution_count": 17,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -347,7 +354,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 17,
+   "execution_count": 18,
    "metadata": {},
    "outputs": [
     {
@@ -377,7 +384,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 18,
+   "execution_count": 19,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -386,7 +393,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 19,
+   "execution_count": 20,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -395,7 +402,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 20,
+   "execution_count": 21,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -405,7 +412,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 21,
+   "execution_count": 22,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -416,7 +423,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 22,
+   "execution_count": 23,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -432,7 +439,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 23,
+   "execution_count": 24,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -441,7 +448,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 24,
+   "execution_count": 25,
    "metadata": {},
    "outputs": [
     {
@@ -450,7 +457,7 @@
        "True"
       ]
      },
-     "execution_count": 24,
+     "execution_count": 25,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -461,7 +468,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 25,
+   "execution_count": 26,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -511,7 +518,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 26,
+   "execution_count": 27,
    "metadata": {
     "scrolled": false
    },
@@ -539,10 +546,10 @@
       "====================================================================================================\n",
       "## Client Performance Stats\n",
       "====================================================================================================\n",
-      "\t - env_creation_wait_time\t => min: 0.0006797313690185547 || mean: 0.006682872772216797 || max: 0.013753414154052734\n",
-      "\t - internal_env_reset_time\t => min: 0.001880645751953125 || mean: 0.0022284984588623047 || max: 0.0025763511657714844\n",
-      "\t - inference_time(approx)\t => min: 2.4080276489257812e-05 || mean: 0.010996600969241616 || max: 2.000370979309082\n",
-      "\t - internal_env_step_time\t => min: 0.0004317760467529297 || mean: 0.0010453842498443946 || max: 0.0029251575469970703\n",
+      "\t - env_creation_wait_time\t => min: 0.0006320476531982422 || mean: 0.01032869021097819 || max: 0.026048660278320312\n",
+      "\t - internal_env_reset_time\t => min: 0.0017096996307373047 || mean: 0.00208437442779541 || max: 0.0024590492248535156\n",
+      "\t - inference_time(approx)\t => min: 2.4318695068359375e-05 || mean: 0.010987052500573663 || max: 2.0010597705841064\n",
+      "\t - internal_env_step_time\t => min: 0.00031685829162597656 || mean: 0.000881081099038596 || max: 0.003511667251586914\n",
       "====================================================================================================\n",
       "{'mean_reward': 1978.0, 'mean_normalized_reward': 0.40367, 'mean_percentage_complete': 0.5}\n",
       "All done.\n"
@@ -567,7 +574,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 27,
+   "execution_count": 28,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -576,7 +583,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 28,
+   "execution_count": 29,
    "metadata": {},
    "outputs": [
     {
@@ -707,8 +714,8 @@
        "    </tr>\n",
        "    <tr>\n",
        "      <th>simulation_time</th>\n",
-       "      <td>0.258108</td>\n",
-       "      <td>2.0388</td>\n",
+       "      <td>0.219365</td>\n",
+       "      <td>2.04248</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>nb_malfunctioning_trains</th>\n",
@@ -717,18 +724,18 @@
        "    </tr>\n",
        "    <tr>\n",
        "      <th>controller_inference_time_min</th>\n",
-       "      <td>2.40803e-05</td>\n",
-       "      <td>2.40803e-05</td>\n",
+       "      <td>2.43187e-05</td>\n",
+       "      <td>2.43187e-05</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>controller_inference_time_mean</th>\n",
-       "      <td>6.7676e-05</td>\n",
-       "      <td>6.59725e-05</td>\n",
+       "      <td>5.34392e-05</td>\n",
+       "      <td>5.2587e-05</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>controller_inference_time_max</th>\n",
-       "      <td>0.000576019</td>\n",
-       "      <td>0.000576019</td>\n",
+       "      <td>0.00029397</td>\n",
+       "      <td>0.00029397</td>\n",
        "    </tr>\n",
        "  </tbody>\n",
        "</table>\n",
@@ -756,14 +763,14 @@
        "normalized_reward                         0.807347                   0\n",
        "percentage_complete                              1                   0\n",
        "steps                                          171                  12\n",
-       "simulation_time                           0.258108              2.0388\n",
+       "simulation_time                           0.219365             2.04248\n",
        "nb_malfunctioning_trains                         2                   0\n",
-       "controller_inference_time_min          2.40803e-05         2.40803e-05\n",
-       "controller_inference_time_mean          6.7676e-05         6.59725e-05\n",
-       "controller_inference_time_max          0.000576019         0.000576019"
+       "controller_inference_time_min          2.43187e-05         2.43187e-05\n",
+       "controller_inference_time_mean         5.34392e-05          5.2587e-05\n",
+       "controller_inference_time_max           0.00029397          0.00029397"
       ]
      },
-     "execution_count": 28,
+     "execution_count": 29,
      "metadata": {},
      "output_type": "execute_result"
     }