diff --git a/code/notebooks/stft.ipynb b/code/notebooks/stft.ipynb
index 0f62cb6..b4b46db 100644
--- a/code/notebooks/stft.ipynb
+++ b/code/notebooks/stft.ipynb
@@ -220,7 +220,7 @@
     "import os\n",
     "import pandas as pd\n",
     "import numpy as np\n",
-    "from scipy.signal import hann\n",
+    "from scipy.signal.windows import hann\n",
     "import multiprocessing"
    ]
   },
@@ -244,16 +244,6 @@
     "Fs = 1024"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "# Define the base directory where DAMAGE_X folders are located\n",
-    "damage_base_path = 'D:/thesis/data/converted/raw'"
-   ]
-  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -282,6 +272,13 @@
     "damage_base_path = 'D:/thesis/data/converted/raw'"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "The path above is specifically intended to process STFT data using Dataset A"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,