Scala: fun sets exercise init