updating tree shaders to be slightly more efficient