It's strange. What you propose works quite well, thank you. However, if I read in that colour data at the beginning of the shader and try to use it at the end of the shader, it doesn't work. Like:
Thanks again. I ended up going with your approach, even though it does not allow me to add a colourful tinge to the AO shading. The reason I went with your approach is because it allows me to draw things like an axis, using the flat colour shader. If I use my old method it does not draw the axis, or it's covered up, or drawn in orange, none of which I wanted. LOL