diff --git a/model_training/xgboost/xgboost.ipynb b/model_training/xgboost/xgboost.ipynb
index 7f8aec0..cdbf64f 100644
--- a/model_training/xgboost/xgboost.ipynb
+++ b/model_training/xgboost/xgboost.ipynb
@@ -2,18 +2,18 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "id": "e3be057e-8d2a-4d05-bd42-6b1dc75df5ed",
    "metadata": {},
    "outputs": [],
    "source": [
-    "import os"
+    "import xgboost as xgb"
    ]
   }
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": "project_fahrsimulator",
    "language": "python",
    "name": "python3"
   },
@@ -27,7 +27,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.10"
+   "version": "3.10.19"
   }
  },
  "nbformat": 4,