updated main.lua