Cleanup simple example, prepare it for docs