diff --git a/.vscode/settings.json b/.vscode/settings.json
new file mode 100644
index 0000000..8bfa42b
--- /dev/null
+++ b/.vscode/settings.json
@@ -0,0 +1,8 @@
+{
+    "files.exclude": {
+    "**/.env": true,
+    "**/__pycache__": true,
+    "**/.ipynb_checkpoints": true,
+    "**/.cache": true
+  }
+}
diff --git a/data.csv b/data/data.csv
similarity index 100%
rename from data.csv
rename to data/data.csv
diff --git a/train.ipynb b/train.ipynb
index ba5c1bd..225ccc6 100644
--- a/train.ipynb
+++ b/train.ipynb
@@ -16,7 +16,7 @@
    "outputs": [],
    "source": [
     "!rm -f tracing_database.json\n",
-    "%pip install great-ai > /dev/null"
+    "%pip install --upgrade great-ai > /dev/null"
    ]
   },
   {
@@ -27,7 +27,7 @@
    "source": [
     "import pandas as pd\n",
     "\n",
-    "data = pd.read_csv('data.csv')\n",
+    "data = pd.read_csv('data/data.csv')\n",
     "\n",
     "pd.set_option(\"display.max_rows\", 30)\n",
     "pd.set_option(\"display.max_columns\", 30)\n",