diff --git a/code/end_to_end_gee/tf_gee_dnn_model.ipynb b/code/end_to_end_gee/tf_gee_dnn_model.ipynb
index 5d0b02bb..1ad1ea52 100644
--- a/code/end_to_end_gee/tf_gee_dnn_model.ipynb
+++ b/code/end_to_end_gee/tf_gee_dnn_model.ipynb
@@ -145,10 +145,10 @@
       "execution_count": 6,
       "metadata": {
         "id": "vJ4U8BdRU4Pw",
-        "outputId": "96b55fbf-23b7-4d3a-c1bf-1502a9bb2ccb",
         "colab": {
           "base_uri": "https://localhost:8080/"
-        }
+        },
+        "outputId": "96b55fbf-23b7-4d3a-c1bf-1502a9bb2ccb"
       },
       "outputs": [
         {
@@ -210,10 +210,10 @@
       "execution_count": 7,
       "metadata": {
         "id": "MiX_T701VW7s",
-        "outputId": "f3bcac7d-dc6c-4495-fe3c-590b5cdac9bf",
         "colab": {
           "base_uri": "https://localhost:8080/"
-        }
+        },
+        "outputId": "f3bcac7d-dc6c-4495-fe3c-590b5cdac9bf"
       },
       "outputs": [
         {
@@ -305,11 +305,11 @@
       "execution_count": 9,
       "metadata": {
         "id": "4v5keB8IPXjZ",
-        "outputId": "84876f2e-a075-422e-a417-7e0e82844f64",
         "colab": {
           "base_uri": "https://localhost:8080/",
           "height": 248
-        }
+        },
+        "outputId": "84876f2e-a075-422e-a417-7e0e82844f64"
       },
       "outputs": [
         {
@@ -425,10 +425,10 @@
       "execution_count": 10,
       "metadata": {
         "id": "U8UrJLp6WKPA",
-        "outputId": "34f62661-31ec-495b-d5de-89f11f74784c",
         "colab": {
           "base_uri": "https://localhost:8080/"
-        }
+        },
+        "outputId": "34f62661-31ec-495b-d5de-89f11f74784c"
       },
       "outputs": [
         {
@@ -602,20 +602,20 @@
     },
     {
       "cell_type": "code",
-      "execution_count": null,
+      "execution_count": 16,
       "metadata": {
         "id": "4rAWJiC5oSdV",
-        "outputId": "95ec60b6-9307-45ca-859c-8324d0adccb1",
         "colab": {
           "base_uri": "https://localhost:8080/"
-        }
+        },
+        "outputId": "95ec60b6-9307-45ca-859c-8324d0adccb1"
       },
       "outputs": [
         {
           "output_type": "stream",
           "name": "stdout",
           "text": [
-            "\u001b[1m  65/1008\u001b[0m \u001b[32m━\u001b[0m\u001b[37m━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[1m35:13\u001b[0m 2s/step"
+            "\u001b[1m1008/1008\u001b[0m \u001b[32m━━━━━━━━━━━━━━━━━━━━\u001b[0m\u001b[37m\u001b[0m \u001b[1m2279s\u001b[0m 2s/step\n"
           ]
         }
       ],