added some code to shader