Fix sample code in readme