Add the ability to update the fragment shader