Simple improvement - added prefix sums.

Signed-off-by: Tomasz Kamiński <kaminski.tomasz.a@gmail.com>