Updated balance method to work correctly. Updated tests to match expected output.