Updated main.lua with display options