Fix readme sample code