Updated assembly code