Updating sample code in readme