diff --git a/code/notebooks/03_feature_extraction.ipynb b/code/notebooks/03_feature_extraction.ipynb
index 5995086..b0de9e1 100644
--- a/code/notebooks/03_feature_extraction.ipynb
+++ b/code/notebooks/03_feature_extraction.ipynb
@@ -29,13 +29,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import sys\n",
     "import numpy as np\n",
-    "import pandas as pd\n",
-    "# Assuming the src directory is one level up from the notebooks directory\n",
-    "sys.path.append('../src/features')\n",
-    "\n",
-    "from time_domain_features import FeatureExtractor"
+    "import pandas as pd\n"
    ]
   },
   {
@@ -276,7 +271,12 @@
    ],
    "source": [
     "import pandas as pd\n",
-    "# Extract features using our class\n",
+    "import sys\n",
+    "# Assuming the src directory is one level up from the notebooks directory\n",
+    "sys.path.append('../src/features')\n",
+    "from time_domain_features import FeatureExtractor\n",
+    "\n",
+    "# Extract features\n",
     "extracted = FeatureExtractor(mock_df['SampleData'])\n",
     "\n",
     "# Format with pandas DataFramw\n",