Add Balance Unit Tests