Update the sample code