Shader update

Not working yet!