Adding PartitionContext class