Pretty main.lua

Prettyfied main.lua a bit
Removed some stuff.
Error to be checked