It would be good to add the capability to use `xarray.Dataset` objects instead of only reading a list of netCDF files as input. This gives flexibility, as `xarray.Dataset`s could be loaded using `Dask`.