Update returns to include the list of IDs