Added tests to verify non null returns in balance calculator