Improved the layout of the example