Add coloring functionality