add documentation for the color function