Add a unfinished version of prefix sum