chore: fix a few cells in the new notebook

[hrodmn]

No description provided.

[github-actions]

Benchmark Comparison

Test	Baseline (ms)	PR (ms)	Change
test_band_access_pattern[multi_band]	1353.2	1094.8	-19.1%
test_band_access_pattern[single_band]	1244.5	1009.0	-18.9%
test_full_compute	640.0	510.2	-20.3%
test_mosaic_method[first]	622.6	510.9	-18.0%
test_mosaic_method[median]	670.1	536.7	-19.9%
test_native_crs_resolution	11441.5	8866.2	-22.5%
test_open_overhead	279.8	236.0	-15.7%
test_reproject_workers[1]	1393.0	1105.1	-20.7%
test_reproject_workers[4]	1397.3	1104.5	-21.0%
test_time_step_parallelism[dask_time_1]	1348.7	1094.5	-18.8%
test_time_step_parallelism[no_dask]	908.6	759.1	-16.5%