Updated AIController