add color wrap function