Add display color function