Update the example code