Updated the generated data to handle partitions