Improve style of the example