Update main.lua