diff --git a/docs/read-write.ipynb b/docs/read-write.ipynb
index 3ae07d1..1666aa9 100644
--- a/docs/read-write.ipynb
+++ b/docs/read-write.ipynb
@@ -89,9 +89,9 @@
           "name": "stdout",
           "output_type": "stream",
           "text": [
-            "Key(offsets={'lat': 0, 'lon': 0, 'time': 0}, vars=None) \u003cclass 'xarray.core.dataset.Dataset'\u003e\n",
-            "Key(offsets={'lat': 0, 'lon': 0, 'time': 1000}, vars=None) \u003cclass 'xarray.core.dataset.Dataset'\u003e\n",
-            "Key(offsets={'lat': 0, 'lon': 0, 'time': 2000}, vars=None) \u003cclass 'xarray.core.dataset.Dataset'\u003e\n"
+            "Key(offsets={'lat': 0, 'lon': 0, 'time': 0}, vars=None) <class 'xarray.core.dataset.Dataset'>\n",
+            "Key(offsets={'lat': 0, 'lon': 0, 'time': 1000}, vars=None) <class 'xarray.core.dataset.Dataset'>\n",
+            "Key(offsets={'lat': 0, 'lon': 0, 'time': 2000}, vars=None) <class 'xarray.core.dataset.Dataset'>\n"
           ]
         }
       ],
@@ -141,7 +141,7 @@
           "name": "stdout",
           "output_type": "stream",
           "text": [
-            "\u003cxarray.Dataset\u003e\n",
+            "<xarray.Dataset>\n",
             "Dimensions:  (time: 2920, lat: 25, lon: 53)\n",
             "Coordinates:\n",
             "  * lat      (lat) float32 75.0 72.5 70.0 67.5 65.0 ... 25.0 22.5 20.0 17.5 15.0\n",
@@ -195,9 +195,9 @@
           "name": "stdout",
           "output_type": "stream",
           "text": [
-            "Key(offsets={'lat': 0, 'lon': 0, 'time': 0}, vars=None) \u003cclass 'xarray.core.dataset.Dataset'\u003e\n",
-            "Key(offsets={'lat': 0, 'lon': 0, 'time': 1000}, vars=None) \u003cclass 'xarray.core.dataset.Dataset'\u003e\n",
-            "Key(offsets={'lat': 0, 'lon': 0, 'time': 2000}, vars=None) \u003cclass 'xarray.core.dataset.Dataset'\u003e\n"
+            "Key(offsets={'lat': 0, 'lon': 0, 'time': 0}, vars=None) <class 'xarray.core.dataset.Dataset'>\n",
+            "Key(offsets={'lat': 0, 'lon': 0, 'time': 1000}, vars=None) <class 'xarray.core.dataset.Dataset'>\n",
+            "Key(offsets={'lat': 0, 'lon': 0, 'time': 2000}, vars=None) <class 'xarray.core.dataset.Dataset'>\n"
           ]
         }
       ],
@@ -245,14 +245,14 @@
           "name": "stdout",
           "output_type": "stream",
           "text": [
-            "\u003cxarray.Dataset\u003e\n",
+            "<xarray.Dataset>\n",
             "Dimensions:  (time: 2920, lat: 25, lon: 53)\n",
             "Coordinates:\n",
             "  * lat      (lat) float32 75.0 72.5 70.0 67.5 65.0 ... 25.0 22.5 20.0 17.5 15.0\n",
             "  * lon      (lon) float32 200.0 202.5 205.0 207.5 ... 322.5 325.0 327.5 330.0\n",
             "  * time     (time) datetime64[ns] 2013-01-01 ... 2014-12-31T18:00:00\n",
             "Data variables:\n",
-            "    air      (time, lat, lon) float32 dask.array\u003cchunksize=(2920, 25, 53), meta=np.ndarray\u003e\n",
+            "    air      (time, lat, lon) float32 dask.array<chunksize=(2920, 25, 53), meta=np.ndarray>\n",
             "Attributes:\n",
             "    Conventions:  COARDS\n",
             "    description:  Data is from NMC initialized reanalysis\\n(4x/day).  These a...\n",
@@ -352,15 +352,15 @@
           "name": "stderr",
           "output_type": "stream",
           "text": [
-            "WARNING:apache_beam.coders.coder_impl:Using fallback deterministic coder for type '\u003cclass 'xarray_beam._src.core.Key'\u003e' in '[11]: ConsolidateChunks/GroupByTempKeys'. \n",
-            "WARNING:apache_beam.coders.coder_impl:Using fallback deterministic coder for type '\u003cclass 'xarray_beam._src.core.Key'\u003e' in '[11]: ConsolidateChunks/GroupByTempKeys'. \n"
+            "WARNING:apache_beam.coders.coder_impl:Using fallback deterministic coder for type '<class 'xarray_beam._src.core.Key'>' in '[11]: ConsolidateChunks/GroupByTempKeys'. \n",
+            "WARNING:apache_beam.coders.coder_impl:Using fallback deterministic coder for type '<class 'xarray_beam._src.core.Key'>' in '[11]: ConsolidateChunks/GroupByTempKeys'. \n"
           ]
         }
       ],
       "source": [
         "all_days = pd.date_range('2013-01-01', '2014-01-01', freq='1D')\n",
         "\n",
-        "def load_one_example(time: pd.Timestamp) -\u003e tuple[xbeam.Key, xarray.Dataset]:\n",
+        "def load_one_example(time: pd.Timestamp) -> tuple[xbeam.Key, xarray.Dataset]:\n",
         "    key = xbeam.Key({'time': (time - all_days[0]).days})\n",
         "    dataset = ds.sel(time=[time])  # replace with your code to create one example\n",
         "    return key, dataset\n",
@@ -428,9 +428,9 @@
           "name": "stdout",
           "output_type": "stream",
           "text": [
-            "Key(offsets={'lat': 0, 'lon': 0, 'time': 0}, vars=None) \u003cclass 'xarray.core.dataset.Dataset'\u003e\n",
-            "Key(offsets={'lat': 0, 'lon': 0, 'time': 1000}, vars=None) \u003cclass 'xarray.core.dataset.Dataset'\u003e\n",
-            "Key(offsets={'lat': 0, 'lon': 0, 'time': 2000}, vars=None) \u003cclass 'xarray.core.dataset.Dataset'\u003e\n"
+            "Key(offsets={'lat': 0, 'lon': 0, 'time': 0}, vars=None) <class 'xarray.core.dataset.Dataset'>\n",
+            "Key(offsets={'lat': 0, 'lon': 0, 'time': 1000}, vars=None) <class 'xarray.core.dataset.Dataset'>\n",
+            "Key(offsets={'lat': 0, 'lon': 0, 'time': 2000}, vars=None) <class 'xarray.core.dataset.Dataset'>\n"
           ]
         }
       ],