Modify AI class