Minor changes to shader