working updateStateByKey