Refactor AIController.addEntity()

Restrict controlled entities too