try to differentiate different class types (colors) .. busted at present