Add color handling