Expand example to use done()