Add running balance (still needs work)