Update main.lua

Improved initial setup