Update momentum-ai.lua