Update mem_alloc.c