diff --git a/datascienceintro/ContinuousCasting_ConicSteelVessel.ipynb b/datascienceintro/ContinuousCasting_ConicSteelVessel.ipynb
index c0fbecd846f1bee99fdf75a8cd483a1a782234e2..8ac02d42d27ac7e26a8de99c181effc4a4380b5b 100644
--- a/datascienceintro/ContinuousCasting_ConicSteelVessel.ipynb
+++ b/datascienceintro/ContinuousCasting_ConicSteelVessel.ipynb
@@ -85,7 +85,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": null,
+      "execution_count": 1,
       "metadata": {
         "id": "pAfdaw0afrWZ"
       },
@@ -691,11 +691,21 @@
       "provenance": []
     },
     "kernelspec": {
-      "display_name": "Python 3",
+      "display_name": "datascienceintro-eVBNPtpL-py3.11",
+      "language": "python",
       "name": "python3"
     },
     "language_info": {
-      "name": "python"
+      "codemirror_mode": {
+        "name": "ipython",
+        "version": 3
+      },
+      "file_extension": ".py",
+      "mimetype": "text/x-python",
+      "name": "python",
+      "nbconvert_exporter": "python",
+      "pygments_lexer": "ipython3",
+      "version": "3.11.11"
     }
   },
   "nbformat": 4,
diff --git a/poetry.lock b/poetry.lock
index 25d53df1d86ace580098c349212f6fa912fb4f0b..46c1fc90d0f1bc40bd9906f63a7667b1690b044a 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -3551,6 +3551,22 @@ files = [
 [package.dependencies]
 torch = "2.6.0"
 
+[[package]]
+name = "torchdiffeq"
+version = "0.2.5"
+description = "ODE solvers and adjoint sensitivity analysis in PyTorch."
+optional = false
+python-versions = "~=3.6"
+groups = ["main"]
+files = [
+    {file = "torchdiffeq-0.2.5-py3-none-any.whl", hash = "sha256:aa1db4bed13bd04952f28a53cdf4336d1ab60417c1d9698d7a239fec1cf2bcf8"},
+    {file = "torchdiffeq-0.2.5.tar.gz", hash = "sha256:b50d3760d13fd138dcceac651f4b80396f44fefcebd037a033fecfeaa9cc12e7"},
+]
+
+[package.dependencies]
+scipy = ">=1.4.0"
+torch = ">=1.5.0"
+
 [[package]]
 name = "torchsummary"
 version = "1.5.1"
@@ -3807,4 +3823,4 @@ test = ["websockets"]
 [metadata]
 lock-version = "2.1"
 python-versions = ">=3.11,<3.12"
-content-hash = "c352f0d569c2f4565ec4e3e279904127a1e8078d3a7118eca86dabe79f1d2de9"
+content-hash = "555ae9e8c489103c8c12b00b839a48bc76ae4e52f01d8752fd75f3634fdec8ab"
diff --git a/pyproject.toml b/pyproject.toml
index 56a893ec85102ce39d654b57545ad55e9ca1dacc..58de0b9bb9f915ae71ad27716e7ec6a884a97fbd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -26,6 +26,7 @@ cyclic-boosting = "^1.4.0"
 ipython = "^9.0.2"
 attrs = "^25.2.0"
 torchsummary = "^1.5.1"
+torchdiffeq = "^0.2.5"
 
 
 [tool.poetry.group.dev.dependencies]
diff --git a/requirements.txt b/requirements.txt
index 553d0a92fdfbd209983020b5a3b8134958ec0d00..4f7cc4766188ac1458503aec59a626b81d7217d8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,6 +10,7 @@ torch==2.6.0
 torchvision==0.21.0
 torchaudio==2.6.0
 torchsummary==1.5.1
+torchdiffeq==0.2.5
 lime==0.2.0.1
 cyclic-boosting==1.4.0