Enhanced example with partition key