update sort colors: simplify code and add problem description