added driver class for testing