small modification to use the color classifier