Update the code of the demo