Update code.

- Adjust layout.