Adding classes and tests for getBalance