Shader class updated

Now it should work properly.