Updated the framework for Test Battles