minor update to the example