Update ProblemTwo.scala