Made changes to main.lua