problem 2 in Scala!!