Improving formatting of sample code for readability