Refactor balance methods and tests.