updated test driver to test the different functions (comments around places that need work)