Code cleanup.

Code cleanup.