diff --git a/examples/network_traffic_prediction/Air_Traffic_Data_Exp&Gc.ipynb b/examples/network_traffic_prediction/Air_Traffic_Data_Exp&Gc.ipynb
index c2944ed9cb..4d69255bf1 100644
--- a/examples/network_traffic_prediction/Air_Traffic_Data_Exp&Gc.ipynb
+++ b/examples/network_traffic_prediction/Air_Traffic_Data_Exp&Gc.ipynb
@@ -534,7 +534,12 @@
     "    \"features\": [\n",
     "        {\n",
     "            \"feature_col\": \"demands\",\n",
-    "            \"feature_name\": \"demands\"\n",
+    "            \"feature_name\": \"demands\",\n",
+    "            \"transform\": {\n",
+    "                        \"name\": \"max_min_norm\",\n",
+    "                        \"max_val\": 10.0,\n",
+    "                        \"min_val\": 0.0\n",
+    "                    }\n",
     "        }\n",
     "    ]\n",
     "}\n",
@@ -555,11 +560,21 @@
     "    \"features\": [\n",
     "        {\n",
     "            \"feature_col\": \"capacity\",\n",
-    "            \"feature_name\": \"capacity\"\n",
+    "            \"feature_name\": \"capacity\",\n",
+    "            \"transform\": {\n",
+    "                        \"name\": \"max_min_norm\",\n",
+    "                        \"max_val\": 10.0,\n",
+    "                        \"min_val\": 0.0\n",
+    "                    }\n",
     "        },\n",
     "        {\n",
     "            \"feature_col\": \"traffics\",\n",
-    "            \"feature_name\": \"traffics\"\n",
+    "            \"feature_name\": \"traffics\",\n",
+    "            \"transform\": {\n",
+    "                        \"name\": \"max_min_norm\",\n",
+    "                        \"max_val\": 10.0,\n",
+    "                        \"min_val\": 0.0\n",
+    "                    }\n",
     "        }\n",
     "    ]\n",
     "}\n",
diff --git a/examples/network_traffic_prediction/Synthetic_Airport_Traffic_wAirlines.ipynb b/examples/network_traffic_prediction/Synthetic_Airport_Traffic_wAirlines.ipynb
index 5335a049a8..eec5f29c36 100644
--- a/examples/network_traffic_prediction/Synthetic_Airport_Traffic_wAirlines.ipynb
+++ b/examples/network_traffic_prediction/Synthetic_Airport_Traffic_wAirlines.ipynb
@@ -154,10 +154,9 @@
     }
    ],
    "source": [
-    "# Extract first 1000 major airports\n",
-    "airports_df = fetch_airports_data(N=1000)\n",
+    "# Extract first 400 major airports\n",
+    "airports_df = fetch_airports_data(N=400)\n",
     "\n",
-    "# Turns out there are only 471 major airports\n",
     "print(airports_df.shape)"
    ]
   },