Update sample code in document