diff --git a/Models/DarkPhoton/DarkPhoton.ipynb b/Models/DarkPhoton/DarkPhoton.ipynb
index 42366f3c..e41d46b4 100644
--- a/Models/DarkPhoton/DarkPhoton.ipynb
+++ b/Models/DarkPhoton/DarkPhoton.ipynb
@@ -264,10 +264,11 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "decay_modes = [\"e_e\", \"mu_mu\", \"pi+_pi-\", \"pi0_gamma\", \"pi+_pi-_pi0\", \"K_K\"] \n",
     "model.set_br_1d(\n",
-    "    modes=[\"e_e\", \"mu_mu\"],\n",
-    "    finalstates=[[11,-11], [13,-13]],\n",
-    "    filenames=[\"model/br/e_e.txt\",\"model/br/mu_mu.txt\"]\n",
+    "    modes = decay_modes,\n",
+    "    finalstates=[[11,-11], [13,-13], [211,-211], [111,22], None, [321,-321]],\n",
+    "    filenames=[\"model/br/\"+mode+\".txt\" for mode in decay_modes],\n",
     ")"
    ]
   },
@@ -833,7 +834,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -847,7 +848,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.10"
+   "version": "3.10.5"
   },
   "toc": {
    "base_numbering": 1,
@@ -867,5 +868,5 @@
   }
  },
  "nbformat": 4,
- "nbformat_minor": 2
+ "nbformat_minor": 4
 }