Update PartitionProblem.java