Каких юнитов вы бы хотели заменить\добавить?

Может быть он контролирует, когда пускать ток, а когда нет. А вообще можно разделить: пусть он больших существ бьет кулаком, а маленких ногой, и ток не помешает.