touch up on the AI