Update DStream.scala