Improve demo (layout)