Add more to the attack logic