Augmented updateStateByKey API

Augment updateStateByKey API by exposing both key and current batch timestamp to applications.
Duplication is removed.