Added color function