from itertools import accumulate, takewhile