Add color picker to UI