Update ch04.scala