Update TestUI.scala