Added simpler version of updateStateByKey API with initialRDD and test.