diff --git a/In_Db2_Machine_Learning/Notebooks/Regression_Demo.ipynb b/In_Db2_Machine_Learning/Notebooks/Regression_Demo.ipynb
index 671e8b6..e7db563 100644
--- a/In_Db2_Machine_Learning/Notebooks/Regression_Demo.ipynb
+++ b/In_Db2_Machine_Learning/Notebooks/Regression_Demo.ipynb
@@ -80,7 +80,7 @@
     "    sys.path.append(module_path)\n",
     "import ibm_db\n",
     "import ibm_db_dbi\n",
-    "import ibm_db_sa\n",
+    "# import ibm_db_sa\n",
     "import pandas as pd\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np"
@@ -1227,4 +1227,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
+}
\ No newline at end of file
diff --git a/In_Db2_Machine_Learning/README.md b/In_Db2_Machine_Learning/README.md
index 22aef0f..6da86a6 100644
--- a/In_Db2_Machine_Learning/README.md
+++ b/In_Db2_Machine_Learning/README.md
@@ -107,7 +107,7 @@ To use the [regression demo](https://github.com/IBM/db2-samples/tree/master/In_D
 - [Numpy](https://pypi.org/project/numpy/)
 - [Matplotlib](https://matplotlib.org/users/installing.html)
 
-Also make sure that you have the [InDBMLModules.py](https://github.com/IBM/db2-samples/tree/master/In_Db2_Machine_Learning/Datasets/InDBMLModules.py) file in the same directory as your notebook.
+Also make sure that you have the [InDBMLModules.py](https://github.com/IBM/db2-samples/tree/master/In_Db2_Machine_Learning/lib/InDBMLModules.py) file in the same directory as your notebook.
 
 Once the above prerequisites have been met, ensure that:
 - The parameters in the connection string variable `conn_str` have been changed to your particular Db2 instance (cell 3)
diff --git a/In_Db2_Machine_Learning/Datasets/InDBMLModules.py b/In_Db2_Machine_Learning/lib/InDBMLModules.py
similarity index 100%
rename from In_Db2_Machine_Learning/Datasets/InDBMLModules.py
rename to In_Db2_Machine_Learning/lib/InDBMLModules.py