Repository for AI class