add color setting function