Added two more functions to "Control AI"