Adds option to use shader in screen space