Added AI controller