Slight adjustment to the shader which should speed it up