Add code in realloc