added test for calling balance method