If sorting by colour works for you, why not use Conditional Formatting to set (or clear) the colour? Pretty sure sorting by colour doesn't care how the colour came to be there.