Update main.lua

Added menu functions