Cleaned up frag shader a bit