Add missing assembly code