Added class Battle