First code for codechef ---- TEST.lua