Added specifying num of partitions.