Fixing readme sample code