Update prefix_sum/main.c