Refactor add IEntityAttack