diff --git a/xai-for-transformer/1-Tutorial_Transformer_Model.ipynb b/xai-for-transformer/1-Tutorial_Transformer_Model.ipynb
index 5ff9493..1d3c814 100644
--- a/xai-for-transformer/1-Tutorial_Transformer_Model.ipynb
+++ b/xai-for-transformer/1-Tutorial_Transformer_Model.ipynb
@@ -52,19 +52,20 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [],
    "source": [
-    "#from google.colab import drive\n",
+    "# Mount drive folder to dbe abale to download repo\n",
+    "# from google.colab import drive\n",
+    "# drive.mount('/content/drive')\n",
     "\n",
-    "#drive.mount('/content/drive')\n",
-    "\n",
-    "#%cd /content/drive/MyDrive\n",
+    "# Switch to correct folder'\n",
+    "# %cd /content/drive/MyDrive\n",
     "\n",
     "# Don't run this cell if you already cloned the repo \n",
-    "#!git clone --branch main https://github.com/HelmholtzAI-Consultants-Munich/XAI-Tutorials.git\n",
-    "#%cd XAI-Tutorials"
+    "# !git clone --branch main https://github.com/HelmholtzAI-Consultants-Munich/XAI-Tutorials.git\n",
+    "# %cd XAI-Tutorials"
    ]
   },
   {
@@ -76,7 +77,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -140,7 +141,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 3,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -175,7 +176,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 4,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -255,7 +256,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 5,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -311,7 +312,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 6,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -357,7 +358,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 7,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -419,7 +420,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 8,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -435,7 +436,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 9,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -462,7 +463,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 10,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -480,7 +481,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 11,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -505,7 +506,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 12,
    "metadata": {},
    "outputs": [
     {
@@ -519,7 +520,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "bc7d082b83534da5b72c046e0229b3ab",
+       "model_id": "98e197577f6044a58af8d8556db555fa",
        "version_major": 2,
        "version_minor": 0
       },
@@ -533,7 +534,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "012054119a834e7a964b4dced2a68006",
+       "model_id": "d106c89d22f444cab972aa2ff1d2c298",
        "version_major": 2,
        "version_minor": 0
       },
@@ -567,7 +568,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 13,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -581,7 +582,7 @@
     "    save_total_limit=3,\n",
     "    num_train_epochs=10,\n",
     "    predict_with_generate=True,\n",
-    "    fp16=True, \n",
+    "    fp16=False, \n",
     "    push_to_hub=False,\n",
     ")"
    ]
@@ -597,7 +598,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": 14,
    "metadata": {},
    "outputs": [],
    "source": [
diff --git a/xai-for-transformer/2-Tutorial_AttentionMaps_Text.ipynb b/xai-for-transformer/2-Tutorial_AttentionMaps_Text.ipynb
index 774fe76..4ac5f87 100644
--- a/xai-for-transformer/2-Tutorial_AttentionMaps_Text.ipynb
+++ b/xai-for-transformer/2-Tutorial_AttentionMaps_Text.ipynb
@@ -56,15 +56,16 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "#from google.colab import drive\n",
+    "# Mount drive folder to dbe abale to download repo\n",
+    "# from google.colab import drive\n",
+    "# drive.mount('/content/drive')\n",
     "\n",
-    "#drive.mount('/content/drive')\n",
-    "\n",
-    "#%cd /content/drive/MyDrive\n",
+    "# Switch to correct folder'\n",
+    "# %cd /content/drive/MyDrive\n",
     "\n",
     "# Don't run this cell if you already cloned the repo \n",
-    "#!git clone --branch main https://github.com/HelmholtzAI-Consultants-Munich/XAI-Tutorials.git\n",
-    "#%cd XAI-Tutorials"
+    "# !git clone --branch main https://github.com/HelmholtzAI-Consultants-Munich/XAI-Tutorials.git\n",
+    "# %cd XAI-Tutorials"
    ]
   },
   {
@@ -508,7 +509,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.4"
+   "version": "3.10.13"
   }
  },
  "nbformat": 4,
diff --git a/xai-for-transformer/3-Tutorial_VIT_Model.ipynb b/xai-for-transformer/3-Tutorial_VIT_Model.ipynb
index ec0cb87..2ee1b18 100644
--- a/xai-for-transformer/3-Tutorial_VIT_Model.ipynb
+++ b/xai-for-transformer/3-Tutorial_VIT_Model.ipynb
@@ -34,7 +34,7 @@
     "If you installed the packages and requirements on your machine, you can skip this section and start from the import section.\n",
     "Otherwise, you can follow and execute the tutorial on your browser. To start working on the notebook, click on the following button. This will open this page in the Colab environment and you will be able to execute the code on your own.\n",
     "\n",
-    "<a href=\"https://colab.research.google.com/github/HelmholtzAI-Consultants-Munich/XAI-Tutorials/blob/main/data_and_models/3-Tutorial_VIT_Model.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+    "<a href=\"https://colab.research.google.com/github/HelmholtzAI-Consultants-Munich/XAI-Tutorials/blob/main/xai-for-transformer/3-Tutorial_VIT_Model.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
    ]
   },
   {
@@ -56,15 +56,16 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "#from google.colab import drive\n",
+    "# Mount drive folder to dbe abale to download repo\n",
+    "# from google.colab import drive\n",
+    "# drive.mount('/content/drive')\n",
     "\n",
-    "#drive.mount('/content/drive')\n",
-    "\n",
-    "#%cd /content/drive/MyDrive\n",
+    "# Switch to correct folder'\n",
+    "# %cd /content/drive/MyDrive\n",
     "\n",
     "# Don't run this cell if you already cloned the repo \n",
-    "#!git clone --branch main https://github.com/HelmholtzAI-Consultants-Munich/XAI-Tutorials.git\n",
-    "#%cd XAI-Tutorials"
+    "# !git clone --branch main https://github.com/HelmholtzAI-Consultants-Munich/XAI-Tutorials.git\n",
+    "# %cd XAI-Tutorials"
    ]
   },
   {
diff --git a/xai-for-transformer/4-Tutorial_AttentionMaps_Images.ipynb b/xai-for-transformer/4-Tutorial_AttentionMaps_Images.ipynb
index 20d2e63..45df3a1 100644
--- a/xai-for-transformer/4-Tutorial_AttentionMaps_Images.ipynb
+++ b/xai-for-transformer/4-Tutorial_AttentionMaps_Images.ipynb
@@ -51,15 +51,16 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "#from google.colab import drive\n",
+    "# Mount drive folder to dbe abale to download repo\n",
+    "# from google.colab import drive\n",
+    "# drive.mount('/content/drive')\n",
     "\n",
-    "#drive.mount('/content/drive')\n",
-    "\n",
-    "#%cd /content/drive/MyDrive\n",
+    "# Switch to correct folder'\n",
+    "# %cd /content/drive/MyDrive\n",
     "\n",
     "# Don't run this cell if you already cloned the repo \n",
-    "#!git clone --branch main https://github.com/HelmholtzAI-Consultants-Munich/XAI-Tutorials.git\n",
-    "#%cd XAI-Tutorials"
+    "# !git clone --branch main https://github.com/HelmholtzAI-Consultants-Munich/XAI-Tutorials.git\n",
+    "# %cd XAI-Tutorials"
    ]
   },
   {