Updated shader code