Add helper function set(). Classic colors. Clean up code.