help me update the code