diff --git a/Train_Colab.ipynb b/Train_Colab.ipynb
index 7b04e3e..68aa8f1 100644
--- a/Train_Colab.ipynb
+++ b/Train_Colab.ipynb
@@ -490,6 +490,9 @@
         "\n",
         "# Finish the training process\n",
         "process.wait()"
+        "if Disconnect_after_training :
+        "  time.sleep(40)
+        "  runtime.unassign()
       ]
     },
     {
@@ -579,4 +582,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
\ No newline at end of file
+}