Add the initial logic for the battle calculator