Minor changes in fragment shader