diff --git a/Train_Colab.ipynb b/Train_Colab.ipynb
index cf118b5..3b901ca 100644
--- a/Train_Colab.ipynb
+++ b/Train_Colab.ipynb
@@ -3,8 +3,8 @@
     {
       "cell_type": "markdown",
       "metadata": {
-        "id": "view-in-github",
-        "colab_type": "text"
+        "colab_type": "text",
+        "id": "view-in-github"
       },
       "source": [
         "<a href=\"https://colab.research.google.com/github/nawnie/EveryDream2trainer/blob/main/Train_Colab.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
@@ -27,8 +27,8 @@
       "cell_type": "code",
       "execution_count": null,
       "metadata": {
-        "id": "WsYIcz9HY9lx",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "WsYIcz9HY9lx"
       },
       "outputs": [],
       "source": [
@@ -121,8 +121,8 @@
       "cell_type": "code",
       "execution_count": null,
       "metadata": {
-        "id": "hAuBbtSvGpau",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "hAuBbtSvGpau"
       },
       "outputs": [],
       "source": [
@@ -177,8 +177,8 @@
       "cell_type": "code",
       "execution_count": null,
       "metadata": {
-        "id": "unaffeqGP_0A",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "unaffeqGP_0A"
       },
       "outputs": [],
       "source": [
@@ -252,23 +252,23 @@
     },
     {
       "cell_type": "markdown",
+      "metadata": {
+        "id": "EHyFzKWXX9kB"
+      },
       "source": [
         "# Training\n",
         "\n",
         "For a more indepth Explanation of each of these paramaters check out /content/EveryDream2trainer/doc.\n",
         "\n",
         "After youve tried a few models you will find /content/EveryDream2trainer/doc/ATWEAKING.md to be extremly helpful."
-      ],
-      "metadata": {
-        "id": "EHyFzKWXX9kB"
-      }
+      ]
     },
     {
       "cell_type": "code",
       "execution_count": null,
       "metadata": {
-        "id": "j9pEI69WXS9w",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "j9pEI69WXS9w"
       },
       "outputs": [],
       "source": [
@@ -492,10 +492,10 @@
         "            last_output = log_line\n",
         "\n",
         "# Finish the training process\n",
-        "process.wait()",
-        "if Disconnect_after_training :",
-        "  time.sleep(40)",
-        "  runtime.unassign()"
+        "process.wait()\n",
+        "if Disconnect_after_training:\n",
+        "    time.sleep(40)\n",
+        "    runtime.unassign()\n"
       ]
     },
     {
@@ -564,8 +564,8 @@
   "metadata": {
     "accelerator": "GPU",
     "colab": {
-      "provenance": [],
-      "include_colab_link": true
+      "include_colab_link": true,
+      "provenance": []
     },
     "gpuClass": "standard",
     "kernelspec": {