Add update_initramfs functionality