Add color option to edit