diff --git a/intro.md b/intro.md
index 8941df7..57bd80c 100644
--- a/intro.md
+++ b/intro.md
@@ -110,4 +110,5 @@ If you find this book useful, please cite it via:
 
 ## Time to get started
 
-The future of simulation is being rewritten, and with the following tools, you’ll be at the forefront of these developments. Let’s dive in!
+The future of simulation is being rewritten, and with the following AI and deep learning techniques, you’ll be at the forefront of these developments. Let’s dive in!
+
diff --git a/probmodels-flowmatching.ipynb b/probmodels-flowmatching.ipynb
index 91d052a..b09c52f 100644
--- a/probmodels-flowmatching.ipynb
+++ b/probmodels-flowmatching.ipynb
@@ -377,7 +377,7 @@
         "import torch\n",
         "from torchdiffeq import odeint\n",
         "\n",
-        "def integrate_flow(model, x0, t_span=(0.0, 1.0), n_steps, method='dopri5'):\n",
+        "def integrate_flow(model, x0, t_span=(0.0, 1.0), n_steps=100, method='dopri5'):\n",
         "\n",
         "    t = torch.linspace(t_span[0], t_span[1], n_steps).to(x0.device)\n",
         "\n",
@@ -393,9 +393,7 @@
         "n_gen = 500\n",
         "\n",
         "x0_gen = torch.randn(n_gen, 2).to(device)\n",
-        "trajectory, time_points = integrate_flow(model, x0_gen, n_steps=100)\n",
-        "\n",
-        "x_gen_final = trajectory[-1].detach().cpu().numpy()"
+        "trajectory, time_points = integrate_flow(model, x0_gen)\n"
       ]
     },
     {
diff --git a/references.bib b/references.bib
index 0130cd8..2513043 100644
--- a/references.bib
+++ b/references.bib
@@ -130,7 +130,7 @@
 @inproceedings{list2022piso,
 	  title={Learned Turbulence Modelling with Differentiable Fluid Solvers},
 	  author={Bjoern List and Liwei Chen and Nils Thuerey},
-	  booktitle={arXiv:2202.06988},
+      booktitle={Journal of Fluid Mechanics (929/25)},
 	  year={2022},
 	  url={https://ge.in.tum.de/publications/},
 }