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