Minor speed improvement with if check
remove print statement
Move old test code to functions, run repeat battles