updated sample code in readme