Created a new Battle class and moved code into the new class.