update Scala code