adding in color key