wrote driver test code