diff --git a/notebooks/data-augmentation/codet-data/Augment_CodeT_codegen.ipynb b/notebooks/data-augmentation/codet-data/Augment_CodeT_codegen.ipynb
index 99fa7ed0..7fc89a06 100644
--- a/notebooks/data-augmentation/codet-data/Augment_CodeT_codegen.ipynb
+++ b/notebooks/data-augmentation/codet-data/Augment_CodeT_codegen.ipynb
@@ -1,5 +1,14 @@
 {
  "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## CodeT Code Generation Datasets\n",
+    "\n",
+    "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/LAION-AI/Open-Assistant/blob/main/notebooks/data-augmentation/codet-data/Augment_CodeT_codegen.ipynb)"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
diff --git a/notebooks/data-augmentation/codet-data/Augment_CodeT_testgen.ipynb b/notebooks/data-augmentation/codet-data/Augment_CodeT_testgen.ipynb
index 0c9f1626..c4641327 100644
--- a/notebooks/data-augmentation/codet-data/Augment_CodeT_testgen.ipynb
+++ b/notebooks/data-augmentation/codet-data/Augment_CodeT_testgen.ipynb
@@ -1,5 +1,14 @@
 {
  "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## CodeT Test Generation Datasets\n",
+    "\n",
+    "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/LAION-AI/Open-Assistant/blob/main/notebooks/data-augmentation/codet-data/Augment_CodeT_testgen.ipynb)"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},