add color function