Optimize `create_xr_metadata`

[lsterzinger]

#4 added a function to add metadata to the xarray dataset, but it's kinda slow. There's probably a good way to speed this up using lower-level xarray API