Need to modify the fragment shader