Pretty main.lua

Prettyfied main.lua a bit