Added the PartitionAwareKey