diff --git a/Druckrohrleitung/Druckstoß_ETH.ipynb b/Druckrohrleitung/Druckstoß_ETH.ipynb
index ff176b1..d7ade0a 100644
--- a/Druckrohrleitung/Druckstoß_ETH.ipynb
+++ b/Druckrohrleitung/Druckstoß_ETH.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -14,7 +14,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -30,13 +30,12 @@
     "n       = 10                # number of pipe segments in discretization\n",
     "nt      = 1500              # number of time steps after initial conditions\n",
     "f_D     = 0.05              # Darcy friction factor\n",
-    "c       = 400               # propagation velocity of the pressure wave [m/s]\n",
-    "\n"
+    "c       = 400               # propagation velocity of the pressure wave [m/s]"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 3,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -64,14 +63,14 @@
     "v_1 = np.zeros_like(t_vec)\n",
     "\n",
     "# storage vector for time evolution of parameters at node N+1 (at valve)\n",
-    "p_np1 = np.zeros_like(t_vec)\n",
-    "v_np1 = np.zeros_like(t_vec)\n",
+    "p_np1 = np.full_like(t_vec,p_old)\n",
+    "v_np1 = np.full_like(t_vec,v_old)\n",
     "\n"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 4,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -92,7 +91,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 5,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -123,14 +122,13 @@
     "    p_old = p_new.copy()\n",
     "    v_old = v_new.copy()\n",
     "\n",
-    "\n",
     "    lo_00.set_ydata(p_new)\n",
     "    lo_01.set_ydata(v_new)\n",
     "    \n",
     "    fig1.suptitle(str(it))\n",
     "    fig1.canvas.draw()\n",
     "    fig1.tight_layout()\n",
-    "    plt.pause(0.1)\n",
+    "    plt.pause(0.01)\n",
     "\n",
     "    # store parameters of node 1 (at reservoir)\n",
     "    p_1[it] = p_new[0]\n",
@@ -139,6 +137,20 @@
     "    p_np1[it] = p_new[-1]\n",
     "    v_np1[it] = v_new[-1]"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "fig2,axs2 = plt.subplots(2,2)\n",
+    "axs2[0,0].plot(t_vec,p_1)\n",
+    "axs2[0,1].plot(t_vec,v_1)\n",
+    "axs2[1,0].plot(t_vec,p_np1)\n",
+    "axs2[1,1].plot(t_vec,v_np1)\n",
+    "plt.show()"
+   ]
   }
  ],
  "metadata": {