Add coloring of arguments