Added better color logic for the lib.