Change update_* signature to take in timestamp

This will help keeping signatures of all update_* functions
same and uniform.