Using functions in shaders