Fix readme code samples