Add classes for battle