Reorganize enemy logic