Added first test for balances