Adding AI module