small changes to ai.