Moves splat code to afterattack

You may now splat cockroaches and butterflies.

code/game/objects/items/weapons/weaponry.dm

@@ -326,3 +326,15 @@
 	attack_verb = list("swatted", "smacked")
 	hitsound = 'sound/effects/snap.ogg'
 	w_class = 2
+
+/obj/item/weapon/melee/flyswatter/afterattack(atom/target, mob/user, proximity_flag)
+	if(proximity_flag)
+		if(istype(target, /mob/living/simple_animal/hostile/poison/bees/) || istype(target, /mob/living/simple_animal/butterfly) || istype(target, /mob/living/simple_animal/cockroach))
+			var/mob/living/simple_animal/bug = target
+			new /obj/effect/decal/cleanable/deadcockroach(get_turf(bug))
+			bug.death(1)
+			user << "<span class='warning'>You easily splat the [bug].</span>"
+		else if(istype(target, /obj/item/queen_bee/))
+			qdel(target)
+			new /obj/effect/decal/cleanable/deadcockroach(get_turf(target))
+			user << "<span class='warning'>You easily splat the [target].</span>"

code/modules/mob/living/simple_animal/hostile/bees.dm

@@ -243,13 +243,6 @@
 		return 1
 	return 0
 
-/mob/living/simple_animal/hostile/poison/bees/attacked_by(obj/item/I, mob/living/user)
-	. = ..()
-	if(istype(I, /obj/item/weapon/melee/flyswatter))
-		user << "<span class='warning'>You easily splat the [src].</span>"
-		new /obj/effect/decal/cleanable/deadcockroach(get_turf(src))
-		death(1)
-
 /obj/item/queen_bee
 	name = "queen bee"
 	desc = "She's the queen of bees, BZZ BZZ!"
@@ -282,10 +275,6 @@
 				name = queen.name
 			else
 				user << "<span class='warning'>You don't have enough units of that chemical to modify the bee's DNA!</span>"
-	if(istype(I,/obj/item/weapon/melee/flyswatter))
-		user << "<span class='warning'>You easily splat the [src].</span>"
-		new /obj/effect/decal/cleanable/deadcockroach(get_turf(src))
-		qdel(src)
 	..()