Add battle function