demo: Improve layout