Updated balance method to work correctly. Updated tests to match expecte...

...d output.