added tests to support balance