Update the code with some changes.