Update 01.answer.scala