Update StatefulNetworkWordCount.scala