ООП Python. Воины

class Warrior:
    HP = 100
    damage = 20
    def attack(self,who):
        who.HP -= self.damage


first_unit = Warrior()
second_unit = Warrior()
while first_unit.HP > 0 and second_unit.HP > 0:
    who_that = int(input('Введите номер юнита, который нанесет удар '))
    if who_that == 1:
        first_unit.attack(second_unit)
        print('Вы атаковали врага,','у него осталось',second_unit.HP,'здоровья')
    else:
        second_unit.attack(first_unit)
        print('Враг атаковал вас,','у вас осталось',first_unit.HP,'здоровья')
if first_unit.HP > second_unit.HP:
    print('Вы победили, ура!')
else:
    print('Вы проиграли, попробуйте снова')