Update the code