Scala assigment week 5