Update Partitioner.scala