Rework mod_initrd function. Add ARM kernel headers in the initrd.