def capitalize_names(ds):
    """
    Capitalize the 'long_name' attribute for plotting purposes
    """
    for var in ds.variables.values():
        if "long_name" in var.attrs:
            var.attrs["long_name"] = var.attrs["long_name"].capitalize()
    return ds