Added structure to main.lua