Updated main driver with test code.