Update main.lua

Fixed menu bug