modify main.lua