ADD selection color function - not finished