Augmented updateStateByKey API

Augment updateStateByKey API by exposing both key and current batch timestamp to applications.
Yep, this was from the fork, should be removed.