Fixing up shaders