Updating main.lua to have options