Update sample code in README