Implement AI mode