Improved formatting.

Improved the code format.