diff --git a/nb_4_full_model.ipynb b/nb_4_full_model.ipynb
index 3388ee0..2a751f2 100644
--- a/nb_4_full_model.ipynb
+++ b/nb_4_full_model.ipynb
@@ -86,44 +86,47 @@
         "# General Details\n",
         "\n",
         "# @markdown ---\n",
-        "# @markdown ### Set the number of triage bays\n",
+        "# @markdown ### Overall Emergency Department Settings\n",
+        "# @markdown #### Set the number of triage bays\n",
         "n_triage = 3 # @param {type:\"slider\", min:1, max:10, step:1}\n",
-        "# @markdown ### Set the probability that a client arriving is a trauma patient (0 = no-one is a trauma patient, 1 = everyone is a trauma patient)\n",
+        "# @markdown #### Set the probability that a client arriving is a trauma patient (0 = no-one is a trauma patient, 1 = everyone is a trauma patient)\n",
         "prob_trauma = 0.3 # @param {type:\"slider\", min:0, max:1, step:0.01}\n",
         "\n",
         "###################################\n",
         "\n",
-        "# Trauma pathway details\n",
+        "\n",
         "# @markdown ---\n",
-        "# @markdown ### Set the number of trauma stabilisation cubicles\n",
+        "# @markdown ### Trauma Pathway Settings\n",
+        "# @markdown #### Set the number of trauma stabilisation cubicles\n",
         "n_trauma = 6 # @param {type:\"slider\", min:1, max:10, step:1}\n",
-        "# @markdown ### Set the number of trauma treatment cubicles\n",
+        "# @markdown #### Set the number of trauma treatment cubicles\n",
         "n_cubicles_2 = 6 # @param {type:\"slider\", min:1, max:10, step:1}\n",
         "\n",
         "###################################\n",
         "\n",
-        "# Non-trauma pathway details\n",
+        "\n",
         "# @markdown ---\n",
-        "# @markdown ### Set the number of non-trauma registration bays\n",
+        "# @markdown ### Non-trauma pathway settings\n",
+        "# @markdown #### Set the number of non-trauma registration bays\n",
         "n_reg = 3 # @param {type:\"slider\", min:1, max:10, step:1}\n",
-        "# @markdown ### Set the number of non-trauma examination rooms\n",
+        "# @markdown #### Set the number of non-trauma examination rooms\n",
         "n_exam = 7 # @param {type:\"slider\", min:1, max:10, step:1}\n",
-        "# @markdown ### Set the number of non-trauma treatment cubicles\n",
+        "# @markdown #### Set the number of non-trauma treatment cubicles\n",
         "n_cubicles_1 = 2  # @param {type:\"slider\", min:1, max:10, step:1} # Probability that a generated non-trauma patient is a trauma patient (set between 0 and 1). DEFAULT: 0.7\n",
-        "# @markdown ### Set the probability that a client arriving is a trauma patient (0 = no-one on the non-trauma pathway needs treatment, 1 = everyone on the non-trauma pathway needs treatment)\n",
+        "# @markdown #### Set the probability that a client arriving is a trauma patient (0 = no-one on the non-trauma pathway needs treatment, 1 = everyone on the non-trauma pathway needs treatment)\n",
         "non_trauma_treat_p = 0.7 # @param {type:\"slider\", min:0, max:1, step:0.01}\n",
         "\n",
         "###################################\n",
         "\n",
         "\n",
         "# @markdown ---\n",
-        "# @markdown ## Model Settings\n",
+        "# @markdown ### Model Settings\n",
         "# @markdown WARNING: Adjusting the n_reps and run_time days upwards will make the model take significantly longer to run.\n",
-        "# @markdown ### Set the random seed\n",
+        "# @markdown #### Set the random seed\n",
         "seed = 42 # @param {type:\"integer\"}\n",
-        "# @markdown ### Set the number of model replications\n",
+        "# @markdown #### Set the number of model replications\n",
         "n_reps = 3 # @param {type:\"slider\", min:0, max:10, step:1}\n",
-        "# @markdown ### Set the number of simulated days the model will run for\n",
+        "# @markdown #### Set the number of simulated days the model will run for\n",
         "run_time_days = 5 # @param {type:\"slider\", min:0, max:30, step:1}"
       ]
     },