updated driver test code