Add the color class
a final change to the controller
use the color class and map results appropriately