forgot the updated fragment shader