Update formation-ai.lua