modified sample code in readme