challenge-4, scala