got ai import working