Update kmalloc.c: small changes