Balance class tests updated, all tests running