Added AI module