challenge-2, scala