Cleaned up the main fragment shader.