Update kmalloc.c: do TODO in kmalloc and fix a mistake