added neatness to main.lua