Added driver test code