Assignment for Scala course