Optimzied the shader