Added more tests of new AI functionality