added driver test code