fragment shader for the square