Added partial kmalloc function