Hacked battle to take either ai_type or ai for an argument.