Fix minor issues for Customer Balance Exercise