Minimal changes to make example work