Minimal changes to the example