Update InterviewSolution.scala