Reworking enemy AI