Not sure which Engine you are using etc., but my general advice for any kind of stylised color palette would be to use Gradient Mapping. It’s very easy to get from any kind of shader black/white effect to a colored effect.
There is an entire post about it here: