Update ConsoleDriver.scala