Update colour shaders to new system via new function,
yet to be integrated into main()