Adding a bit more logic to defense.