change logic of balance calculate