Scala exercise initial draft