add color handling