working on fixing ai.