AI class update