added some stuff to the shader