Improve the example UI some