Update main.lua (2)