added more code to shader