Update State.scala