diff --git a/docs/images/feature_flow.png b/docs/images/feature_flow.png
index 4e82aa6fa..2d408c87a 100644
Binary files a/docs/images/feature_flow.png and b/docs/images/feature_flow.png differ
diff --git a/docs/samples/nyc_taxi_demo.ipynb b/docs/samples/nyc_taxi_demo.ipynb
index c9eb11bba..dc47b8ea8 100644
--- a/docs/samples/nyc_taxi_demo.ipynb
+++ b/docs/samples/nyc_taxi_demo.ipynb
@@ -625,6 +625,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {
     "application/vnd.databricks.v1+cell": {
@@ -637,7 +638,7 @@
    "source": [
     "#### Define derived features\n",
     "\n",
-    "We also define a derived feature, `f_trip_time_distance`, from the anchored features `f_trip_distance` and `f_trip_time_duration` as follows:"
+    "We also define a derived feature, `f_trip_speed`, from the anchored features `f_trip_distance` and `f_trip_time_duration` as follows:"
    ]
   },
   {
@@ -655,7 +656,7 @@
    "source": [
     "derived_features = [\n",
     "    DerivedFeature(\n",
-    "        name=\"f_trip_time_distance\",\n",
+    "        name=\"f_trip_speed\",\n",
     "        feature_type=FLOAT,\n",
     "        input_features=[\n",
     "            f_trip_distance,\n",
@@ -1154,11 +1155,11 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.8 (main, Nov 24 2022, 14:13:03) [GCC 11.2.0]"
+   "version": "3.8.0 (default, Nov  6 2019, 21:49:08) \n[GCC 7.3.0]"
   },
   "vscode": {
    "interpreter": {
-    "hash": "e34a1a57d2e174682770a82d94a178aa36d3ccfaa21227c5d2308e319b7ae532"
+    "hash": "ddb0e38f168d5afaa0b8ab4851ddd8c14364f1d087c15de6ff2ee5a559aec1f2"
    }
   }
  },