Add code update handling.