Fix sample code format