Refactored test driver
Moved color settings to the test driver