fix balance reporting issues - with unit tests!