Refactor on finding enemies