def METHOD_NAME(census_connection, query_data):
    logger.info(
    "Adding extented metadata to gene expression data."
    )
    census_datasets = census_connection["census_info"]["datasets"].read().concat().to_pandas()
    query_data.obs.dataset_id = query_data.obs.dataset_id.astype("category")
    dataset_info = census_datasets[census_datasets.dataset_id.isin(query_data.obs.dataset_id.cat.categories)]\
    [['collection_id', 'collection_name', 'collection_doi', 'dataset_id', 'dataset_title']]\
    .reset_index(drop=True)\
    .apply(lambda x: x.astype('category'))
    return query_data.obs.merge(
        dataset_info, on='dataset_id', how = 'left'
        )