Modified example to match new script