Added more AI testing functionality.