Minimal changes to make example work

These changes make the example script work.