Added AI framework