Add arena_restore