diff --git a/component_charts.ipynb b/component_charts.ipynb
index d87477a..f805ef1 100644
--- a/component_charts.ipynb
+++ b/component_charts.ipynb
@@ -55,8 +55,9 @@
     {
       "cell_type": "code",
       "source": [
-        "import altair as alt\n",
+        "# @title Chart functions { display-mode: \"form\" }\n",
         "\n",
+        "import altair as alt\n",
         "\n",
         "class MissingColumnsError(Exception):\n",
         "    def __init__(self, columns):\n",
diff --git a/template_data_quality_feedback.ipynb b/template_data_quality_feedback.ipynb
index 4f348c5..15dbb3b 100644
--- a/template_data_quality_feedback.ipynb
+++ b/template_data_quality_feedback.ipynb
@@ -254,8 +254,9 @@
         }
       ],
       "source": [
-        "import altair as alt\n",
+        "# @title Chart functions { display-mode: \"form\" }\n",
         "\n",
+        "import altair as alt\n",
         "\n",
         "class MissingColumnsError(Exception):\n",
         "    def __init__(self, columns):\n",
diff --git a/template_publisher_analysis.ipynb b/template_publisher_analysis.ipynb
index cbdb5c3..76725ac 100644
--- a/template_publisher_analysis.ipynb
+++ b/template_publisher_analysis.ipynb
@@ -254,8 +254,9 @@
         }
       ],
       "source": [
-        "import altair as alt\n",
+        "# @title Chart functions { display-mode: \"form\" }\n",
         "\n",
+        "import altair as alt\n",
         "\n",
         "class MissingColumnsError(Exception):\n",
         "    def __init__(self, columns):\n",
diff --git a/template_structure_and_format_feedback.ipynb b/template_structure_and_format_feedback.ipynb
index 97b15d5..df6963b 100644
--- a/template_structure_and_format_feedback.ipynb
+++ b/template_structure_and_format_feedback.ipynb
@@ -254,8 +254,9 @@
         }
       ],
       "source": [
-        "import altair as alt\n",
+        "# @title Chart functions { display-mode: \"form\" }\n",
         "\n",
+        "import altair as alt\n",
         "\n",
         "class MissingColumnsError(Exception):\n",
         "    def __init__(self, columns):\n",
diff --git a/template_usability_checks.ipynb b/template_usability_checks.ipynb
index c2e4191..1516ebf 100644
--- a/template_usability_checks.ipynb
+++ b/template_usability_checks.ipynb
@@ -254,8 +254,9 @@
         }
       ],
       "source": [
-        "import altair as alt\n",
+        "# @title Chart functions { display-mode: \"form\" }\n",
         "\n",
+        "import altair as alt\n",
         "\n",
         "class MissingColumnsError(Exception):\n",
         "    def __init__(self, columns):\n",
diff --git a/template_usability_checks_fieldlist.ipynb b/template_usability_checks_fieldlist.ipynb
index e887dc9..7baae21 100644
--- a/template_usability_checks_fieldlist.ipynb
+++ b/template_usability_checks_fieldlist.ipynb
@@ -254,8 +254,9 @@
         }
       ],
       "source": [
-        "import altair as alt\n",
+        "# @title Chart functions { display-mode: \"form\" }\n",
         "\n",
+        "import altair as alt\n",
         "\n",
         "class MissingColumnsError(Exception):\n",
         "    def __init__(self, columns):\n",
diff --git a/template_usability_checks_registry.ipynb b/template_usability_checks_registry.ipynb
index 9bd27ca..91920fb 100644
--- a/template_usability_checks_registry.ipynb
+++ b/template_usability_checks_registry.ipynb
@@ -254,8 +254,9 @@
         }
       ],
       "source": [
-        "import altair as alt\n",
+        "# @title Chart functions { display-mode: \"form\" }\n",
         "\n",
+        "import altair as alt\n",
         "\n",
         "class MissingColumnsError(Exception):\n",
         "    def __init__(self, columns):\n",