major changes to get color classification working