Fix readme code 2