Fix sample code layout