Multiple Choice Grid - But with Colours

Hello!
I am working on a survey that will be completed on mobile phones and I would like to use the drag and drop, but unfortunately it is WAY too glitchy to use with a small screen.
I am wondering if there is a way to have respondents use three different colors to sort multiple choice options into categories.
This would be a blend of the multiple choice and highlighter questions.

Or even something like this using Dragula...
https://codepen.io/chambaz/pen/yJBXrY

Really appreciate your help!
Jess