Added AI capabilities