updated color classes and added driver class