NFAA fragment shader cleaned up a bit