Update main.lua

Added a little bit of code for future expansion.