Add AI module (WIP)