Modify the example to be simple to reuse