Updated main.lua