remove the println. and add an assert on total number of partitions?