Update test file

Update test file to call the right functions and objects.