diff --git a/EveryDream_Tools.ipynb b/EveryDream_Tools.ipynb
index 8feffa5..b10e284 100644
--- a/EveryDream_Tools.ipynb
+++ b/EveryDream_Tools.ipynb
@@ -10,6 +10,19 @@
         "<a href=\"https://colab.research.google.com/github/victorchall/EveryDream/blob/main/EveryDream_Tools.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
       ]
     },
+    {
+      "cell_type": "code",
+      "source": [
+        "#@title #Connect to Google Drive\n",
+        "from google.colab import drive\n",
+        "drive.mount('/content/drive')\n"
+      ],
+      "metadata": {
+        "id": "Z_ZHfnQ52dg9"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
     {
       "cell_type": "markdown",
       "metadata": {
@@ -57,7 +70,8 @@
         "!pip install fairscale=='0.4.4'\n",
         "!pip install transformers=='4.19.2'\n",
         "!pip install timm\n",
-        "!pip install aiofiles"
+        "!pip install aiofiles\n",
+        "!pip install colorama"
       ]
     },
     {
@@ -275,7 +289,7 @@
     {
       "cell_type": "code",
       "source": [
-        "/content/EveryDream/scripts/compress_img.py \\\n",
+        "!python scripts/compress_img.py \\\n",
         "--img_dir output \\\n",
         "--out_dir output/compressed_images \\\n",
         "--max_mp 1.5 \n",
@@ -309,8 +323,6 @@
       },
       "outputs": [],
       "source": [
-        "from google.colab import drive\n",
-        "drive.mount('/content/drive')\n",
         "\n",
         "!mkdir /content/drive/MyDrive/Auto_Data_sets\n",
         "!cp -r output/ /content/drive/MyDrive/Auto_Data_sets"
@@ -350,6 +362,7 @@
   "metadata": {
     "colab": {
       "provenance": [],
+      "machine_shape": "hm",
       "include_colab_link": true
     },
     "kernelspec": {
@@ -365,7 +378,9 @@
       "interpreter": {
         "hash": "faf4a6abb601e3a9195ce3e9620411ceec233a951446de834cdf28542d2d93b4"
       }
-    }
+    },
+    "accelerator": "GPU",
+    "gpuClass": "standard"
   },
   "nbformat": 4,
   "nbformat_minor": 0