diff --git a/Generate Output.ipynb b/Generate Output.ipynb
index 41ff5a340e054585fd211e64042d5e0e87c34340..9b6f6f5d60d0e8edf07dacae128a9121e2071b78 100644
--- a/Generate Output.ipynb	
+++ b/Generate Output.ipynb	
@@ -30,6 +30,13 @@
     "%matplotlib inline"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Loading the pickle file "
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 4,
@@ -43,6 +50,13 @@
     "    clf = pickle.load(file)"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Reading the real file "
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 8,
@@ -53,25 +67,30 @@
     "X_test = pd.read_csv('DataFiles/7ea49f78-249c-471f-82fd-b2876ba8b10d_cleaned.csv', sep=',', low_memory=False) "
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Predicting the outcome"
+   ]
+  },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "C:\\Users\\devel\\AppData\\Roaming\\Python\\Python312\\site-packages\\sklearn\\base.py:486: UserWarning: X has feature names, but KNeighborsClassifier was fitted without feature names\n",
-      "  warnings.warn(\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "y_predict = clf.predict(X_test)\n",
     "X_test['Output']=y_predict"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Export the data to a CSV for evaluation"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 11,
diff --git a/KNN.ipynb b/KNN.ipynb
index 7567b7e76ad9194da080aefc8cf0b4456f354f70..0645fad8995f51e705a5e78f229a331e7f79e151 100644
--- a/KNN.ipynb
+++ b/KNN.ipynb
@@ -1121,8 +1121,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import pickle\n",
-    "\n"
+    "import pickle"
    ]
   },
   {