Update the code in the example