Minor update to the example