diff --git a/.ipynb_checkpoints/multi-area-model-checkpoint.ipynb b/.ipynb_checkpoints/multi-area-model-checkpoint.ipynb
index 44c9aadd085fce1067b633b03191ac60f225e083..28afbcb05c1ba2ea450016a2a922fc464f5e80af 100644
--- a/.ipynb_checkpoints/multi-area-model-checkpoint.ipynb
+++ b/.ipynb_checkpoints/multi-area-model-checkpoint.ipynb
@@ -1358,7 +1358,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 23,
+   "execution_count": 24,
    "id": "bea30fc8",
    "metadata": {},
    "outputs": [
@@ -1369,7 +1369,7 @@
      "traceback": [
       "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
       "\u001b[0;31mAttributeError\u001b[0m                            Traceback (most recent call last)",
-      "Input \u001b[0;32mIn [23]\u001b[0m, in \u001b[0;36m<module>\u001b[0;34m\u001b[0m\n\u001b[1;32m      4\u001b[0m rate \u001b[38;5;241m=\u001b[39m spikecount \u001b[38;5;241m/\u001b[39m M\u001b[38;5;241m.\u001b[39msimulation\u001b[38;5;241m.\u001b[39mparams[\u001b[38;5;124m'\u001b[39m\u001b[38;5;124mdt\u001b[39m\u001b[38;5;124m'\u001b[39m] \u001b[38;5;241m*\u001b[39m \u001b[38;5;241m1e3\u001b[39m \u001b[38;5;241m/\u001b[39m np\u001b[38;5;241m.\u001b[39msum(M\u001b[38;5;241m.\u001b[39mN_vec)\n\u001b[1;32m      6\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mMAM2EBRAINS\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m plot_instan_mean_firing_rate\n\u001b[0;32m----> 7\u001b[0m \u001b[43mplot_instan_mean_firing_rate\u001b[49m\u001b[43m(\u001b[49m\u001b[43mtsteps\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mrate\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43msim_params\u001b[49m\u001b[43m)\u001b[49m\n",
+      "Input \u001b[0;32mIn [24]\u001b[0m, in \u001b[0;36m<module>\u001b[0;34m\u001b[0m\n\u001b[1;32m      4\u001b[0m rate \u001b[38;5;241m=\u001b[39m spikecount \u001b[38;5;241m/\u001b[39m M\u001b[38;5;241m.\u001b[39msimulation\u001b[38;5;241m.\u001b[39mparams[\u001b[38;5;124m'\u001b[39m\u001b[38;5;124mdt\u001b[39m\u001b[38;5;124m'\u001b[39m] \u001b[38;5;241m*\u001b[39m \u001b[38;5;241m1e3\u001b[39m \u001b[38;5;241m/\u001b[39m np\u001b[38;5;241m.\u001b[39msum(M\u001b[38;5;241m.\u001b[39mN_vec)\n\u001b[1;32m      6\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mMAM2EBRAINS\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m plot_instan_mean_firing_rate\n\u001b[0;32m----> 7\u001b[0m \u001b[43mplot_instan_mean_firing_rate\u001b[49m\u001b[43m(\u001b[49m\u001b[43mtsteps\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mrate\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43msim_params\u001b[49m\u001b[43m)\u001b[49m\n",
       "File \u001b[0;32m~/MAM2EBRAINS/./figures/MAM2EBRAINS.py:22\u001b[0m, in \u001b[0;36mplot_instan_mean_firing_rate\u001b[0;34m(tsteps, rate, sim_params)\u001b[0m\n\u001b[1;32m     20\u001b[0m \u001b[38;5;28;01mdef\u001b[39;00m \u001b[38;5;21mplot_instan_mean_firing_rate\u001b[39m(tsteps, rate, sim_params):\n\u001b[1;32m     21\u001b[0m     ax \u001b[38;5;241m=\u001b[39m pl\u001b[38;5;241m.\u001b[39msubplots()\n\u001b[0;32m---> 22\u001b[0m     \u001b[43max\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mplot\u001b[49m(tsteps, rate)\n\u001b[1;32m     23\u001b[0m     ax\u001b[38;5;241m.\u001b[39mplot(tsteps, np\u001b[38;5;241m.\u001b[39maverage(rate)\u001b[38;5;241m*\u001b[39mnp\u001b[38;5;241m.\u001b[39mones(\u001b[38;5;28mlen\u001b[39m(tsteps)), label\u001b[38;5;241m=\u001b[39m\u001b[38;5;124m'\u001b[39m\u001b[38;5;124mmean\u001b[39m\u001b[38;5;124m'\u001b[39m)\n\u001b[1;32m     24\u001b[0m     ax\u001b[38;5;241m.\u001b[39mset_title(\u001b[38;5;124m'\u001b[39m\u001b[38;5;124mInstantaneous and mean firing rate across all populations\u001b[39m\u001b[38;5;124m'\u001b[39m)\n",
       "\u001b[0;31mAttributeError\u001b[0m: 'tuple' object has no attribute 'plot'"
      ]
diff --git a/multi-area-model.ipynb b/multi-area-model.ipynb
index 44c9aadd085fce1067b633b03191ac60f225e083..28afbcb05c1ba2ea450016a2a922fc464f5e80af 100644
--- a/multi-area-model.ipynb
+++ b/multi-area-model.ipynb
@@ -1358,7 +1358,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 23,
+   "execution_count": 24,
    "id": "bea30fc8",
    "metadata": {},
    "outputs": [
@@ -1369,7 +1369,7 @@
      "traceback": [
       "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
       "\u001b[0;31mAttributeError\u001b[0m                            Traceback (most recent call last)",
-      "Input \u001b[0;32mIn [23]\u001b[0m, in \u001b[0;36m<module>\u001b[0;34m\u001b[0m\n\u001b[1;32m      4\u001b[0m rate \u001b[38;5;241m=\u001b[39m spikecount \u001b[38;5;241m/\u001b[39m M\u001b[38;5;241m.\u001b[39msimulation\u001b[38;5;241m.\u001b[39mparams[\u001b[38;5;124m'\u001b[39m\u001b[38;5;124mdt\u001b[39m\u001b[38;5;124m'\u001b[39m] \u001b[38;5;241m*\u001b[39m \u001b[38;5;241m1e3\u001b[39m \u001b[38;5;241m/\u001b[39m np\u001b[38;5;241m.\u001b[39msum(M\u001b[38;5;241m.\u001b[39mN_vec)\n\u001b[1;32m      6\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mMAM2EBRAINS\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m plot_instan_mean_firing_rate\n\u001b[0;32m----> 7\u001b[0m \u001b[43mplot_instan_mean_firing_rate\u001b[49m\u001b[43m(\u001b[49m\u001b[43mtsteps\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mrate\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43msim_params\u001b[49m\u001b[43m)\u001b[49m\n",
+      "Input \u001b[0;32mIn [24]\u001b[0m, in \u001b[0;36m<module>\u001b[0;34m\u001b[0m\n\u001b[1;32m      4\u001b[0m rate \u001b[38;5;241m=\u001b[39m spikecount \u001b[38;5;241m/\u001b[39m M\u001b[38;5;241m.\u001b[39msimulation\u001b[38;5;241m.\u001b[39mparams[\u001b[38;5;124m'\u001b[39m\u001b[38;5;124mdt\u001b[39m\u001b[38;5;124m'\u001b[39m] \u001b[38;5;241m*\u001b[39m \u001b[38;5;241m1e3\u001b[39m \u001b[38;5;241m/\u001b[39m np\u001b[38;5;241m.\u001b[39msum(M\u001b[38;5;241m.\u001b[39mN_vec)\n\u001b[1;32m      6\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mMAM2EBRAINS\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m plot_instan_mean_firing_rate\n\u001b[0;32m----> 7\u001b[0m \u001b[43mplot_instan_mean_firing_rate\u001b[49m\u001b[43m(\u001b[49m\u001b[43mtsteps\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mrate\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43msim_params\u001b[49m\u001b[43m)\u001b[49m\n",
       "File \u001b[0;32m~/MAM2EBRAINS/./figures/MAM2EBRAINS.py:22\u001b[0m, in \u001b[0;36mplot_instan_mean_firing_rate\u001b[0;34m(tsteps, rate, sim_params)\u001b[0m\n\u001b[1;32m     20\u001b[0m \u001b[38;5;28;01mdef\u001b[39;00m \u001b[38;5;21mplot_instan_mean_firing_rate\u001b[39m(tsteps, rate, sim_params):\n\u001b[1;32m     21\u001b[0m     ax \u001b[38;5;241m=\u001b[39m pl\u001b[38;5;241m.\u001b[39msubplots()\n\u001b[0;32m---> 22\u001b[0m     \u001b[43max\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mplot\u001b[49m(tsteps, rate)\n\u001b[1;32m     23\u001b[0m     ax\u001b[38;5;241m.\u001b[39mplot(tsteps, np\u001b[38;5;241m.\u001b[39maverage(rate)\u001b[38;5;241m*\u001b[39mnp\u001b[38;5;241m.\u001b[39mones(\u001b[38;5;28mlen\u001b[39m(tsteps)), label\u001b[38;5;241m=\u001b[39m\u001b[38;5;124m'\u001b[39m\u001b[38;5;124mmean\u001b[39m\u001b[38;5;124m'\u001b[39m)\n\u001b[1;32m     24\u001b[0m     ax\u001b[38;5;241m.\u001b[39mset_title(\u001b[38;5;124m'\u001b[39m\u001b[38;5;124mInstantaneous and mean firing rate across all populations\u001b[39m\u001b[38;5;124m'\u001b[39m)\n",
       "\u001b[0;31mAttributeError\u001b[0m: 'tuple' object has no attribute 'plot'"
      ]