Added some AI testing