Small shader modifications