Add main method that simulates a battle