refactering test driver