put in framework for ai