added some tests in driver