Making lighting frag shader easier to understand