Actually, you could just use the color ramp for that... Guess I've been doing too much with shaders lately XD.
