Improve formatting of example code