diff --git a/experimentation/Diabetes Ridge Regression Training.ipynb b/experimentation/Diabetes Ridge Regression Training.ipynb
index 12fbc6c1..23dae9d1 100644
--- a/experimentation/Diabetes Ridge Regression Training.ipynb	
+++ b/experimentation/Diabetes Ridge Regression Training.ipynb	
@@ -62,19 +62,13 @@
     "reg_model = Ridge(**args)"
    ]
    },
-   {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "## Split the dataframe into test and train data"
-   ]
-  },
    {
    "cell_type": "code",
    "execution_count": 10,
    "metadata": {},
    "outputs": [],
    "source": [
+   "# Split the dataframe into test and train data\n",  
     "def = split_data(df):\n",
     "X = df.drop('Y', axis=1).values\n",
     "y = df['Y'].values\n",