improve colors code