Shader code set up