add argument for color