Update some code.

Update some code.